T4.2_Domain_Shift_KPIs/setup.py at main · AI4REALNET/T4.2_Domain_Shift_KPIs · 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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import os
import setuptools
from setuptools import setup
from domain_shift_kpis import get_version

def get_data_files(directory):
    my_list = []
    for dirpath,_,filenames in os.walk(directory):
        for f in filenames:
            my_list.append((dirpath, [os.path.join(dirpath,f)]))
    return my_list

pkgs = {
    "required": [
        "numpy==2.2.6",
        "gymnasium==1.2.1",
    ],
    "extras": {
        "powergrid": [
            "grid2op==1.12.1",
            "pybind11==2.8.1",
            "lightsim2grid==0.10.3",
            "numba==0.62.1",
            "protobuf==6.33.0",
            "pandapower==3.2.0",
            "pandas==2.3.3",
            "jupyter",
            "lxml==6.0.2",
            "tensorflow==2.8.1",
            "torch==2.0.1",
            "imageio==2.34.0",
            "plotly==5.20.0",
            "tensorboard==2.20.0",
            "l2rpn_baselines==0.8.0",
            "stable-baselines3==2.7.0",
            "stable-baselines3[extra]",
        ],
        "railway": [
            "Flatland"
        ],
        "atm": [
            "BlueSky"
        ],
        "test": [
            "pytest",
            "pytest-cov",
            "pytest-html",
            "pytest-metadata",
            "ipykernel",
            "pylint",
            "pylint-exit",
            "jupytext"
        ],
        "docker": [
            "grid2op==1.12.1",
            "pybind11==2.8.1",
            "lightsim2grid==0.10.3",
            "numba==0.62.1",
            "protobuf==6.33.0",
            "pandapower==3.2.0",
            "pandas==2.3.3",
            "jupyter",
            "lxml==6.0.2",
            "tensorboard==2.20.0",
            "l2rpn_baselines==0.8.0",
            "stable-baselines3==2.7.0",
            "stable-baselines3[extra]",
        ]
    }
}

pkgs["extras"]["test"] += pkgs["required"]

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

setup(name='domain_shift_kpis',
      version=get_version("__init__.py"),
      description='Domain shift KPIs',
      long_description=long_description,
      long_description_content_type="text/markdown",
      classifiers=[
          'Development Status :: 4 - Beta',
          'Programming Language :: Python :: 3.8',
          'Programming Language :: Python :: 3.9',
          'Programming Language :: Python :: 3.10',
          "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
          "Intended Audience :: Developers",
          "Intended Audience :: Education",
          "Intended Audience :: Science/Research",
          "Natural Language :: English"
      ],
      keywords='Domain shift, KPIs, Adaptation time, RL',
      author='Milad Leyli-abadi',
      author_email='milad.leyli-abadi@irt-systemx.fr',
      url="https://github.com/AI4REALNET/T4.2_Domain_Shift_KPIs.git",
      license='MPL',
      packages=setuptools.find_packages(),
      include_package_data=True,
      package_data={
            # If any package contains *.txt or *.rst files, include them:
            "": ["*.ini"],
            },
    #   data_files=get_data_files("configurations"),#[("configurations/powergrid/benchmarks/", ["configurations/powergrid/benchmarks/l2rpn_case14_sandbox.ini"])],
      install_requires=pkgs["required"],
      extras_require=pkgs["extras"],
      zip_safe=False,
      entry_points={
          'console_scripts': []
     }
)