Updated setup and minimal dependencies

Author: ishevche

setup.py

@@ -33,9 +33,9 @@ def get_long_description() -> str:
     # "Programming Language :: Python :: 3.11",
     # "Programming Language :: Python :: 3.12",
 ]
-INSTALL_REQUIRES = (min_deps.tag_to_packages["metrics"],)
+INSTALL_REQUIRES = (min_deps.tag_to_packages["install"],)
 EXTRAS_REQUIRE = {
-    key: value for key, value in min_deps.tag_to_packages.items() if key != "metrics"
+    key: value for key, value in min_deps.tag_to_packages.items() if key != "install"
 }
 
 setup(

xai_ranking/_min_dependencies.py

@@ -4,16 +4,20 @@
 
 # The values are (version_spec, comma separated tags)
 dependent_packages = {
-    "numpy": ("1.20.0", "metrics, datasets, scores"),
-    "pandas": ("1.3.5", "metrics, datasets"),
-    "scipy": ("1.14.1", "metrics"),
-    "scikit-learn": ("1.2.0", "metrics"),
+    "numpy": ("1.20.0", "install"),
+    "pandas": ("1.3.5", "install"),
+    "scipy": ("1.14.1", "install"),
+    "scikit-learn": ("1.2.0", "install"),
     "pytest-cov": ("3.0.0", "tests"),
     "flake8": ("3.8.2", "tests"),
     "black": ("22.3", "tests"),
     "pylint": ("2.12.2", "tests"),
     "mypy": ("1.6.1", "tests"),
     "sphinx": ("4.2.0", "docs"),
+    "statsmodels": ("0.14.4", "optional"),
+    "lime": ("0.2.0.1", "optional"),
+    "shap": ("0.46.0", "optional"),
+    "xai-sharp": ("0.1a1", "optional"),
     # dev
     # "coverage": ("", "tests"),
     # "click": ("", "tests"),
@@ -23,30 +27,14 @@
     # L2R
     # "lightgbm" : ("", "install"),
     # general?
-    # "xai-sharp": ("0.1.a1", "install"),
-    # "shap" : ("", "install"),
-    # "lime" : ("", "install"),
-    # "statsmodels" : ("", "install"),
     # "ml-research" : ("", "install"),
     # dataset module
     # "openpyxl" : ("", "install"),
-    # "" : ("", "install"),
 }
 
 # create inverse mapping for setuptools
 tag_to_packages: dict = {
-    extra: []
-    for extra in [
-        "install",
-        "optional",
-        "docs",
-        "examples",
-        "tests",
-        "all",
-        "metrics",
-        "datasets",
-        "scores",
-    ]
+    extra: [] for extra in ["install", "optional", "docs", "examples", "tests", "all"]
 }
 for package, (min_version, extras) in dependent_packages.items():
     for extra in extras.split(", "):