nle_toolbox/setup.py at main · ivannz/nle_toolbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os
from setuptools import setup


if __name__ == "__main__":
    # update the version number
    version = open("VERSION", "r").read().strip()

    cwd = os.path.dirname(os.path.abspath(__file__))
    with open(os.path.join(cwd, "nle_toolbox", "__version__.py"), "w") as f:
        f.write(f"__version__ = '{version}'\n")

    setup(
        name="nle_toolbox",
        version=version,
        description="""Toolbox for NLE Challenge""",
        long_description=open("README.md").read(),
        long_description_content_type="text/markdown",
        license="MIT",
        packages=[
            "nle_toolbox",
            "nle_toolbox.utils",
            "nle_toolbox.utils.env",
            "nle_toolbox.utils.env.minihack",
            "nle_toolbox.utils.replay",
            "nle_toolbox.utils.rl",
            "nle_toolbox.bot",
            "nle_toolbox.bot.model",
            "nle_toolbox.zoo",
            "nle_toolbox.zoo.models",
            "nle_toolbox.zoo.transformer",
            "nle_toolbox.zoo.vq",
        ],
        python_requires=">=3.9",
        install_requires=[
            "numpy",
            "torch>=1.8",
            "python-plyr>=0.8",
            "scipy",
            "einops",
            "gym",
            "nle>=0.8.1",
            "minihack",
            "matplotlib",
        ],
        test_requires=[
            "gitpython",
            "pytest",
            "swig",
            "box2d-py",
            "gym[box2d]",
        ],
    )