diff --git a/aif360/algorithms/__init__.py b/aif360/algorithms/__init__.py index 20cc605f..6b2a9d3f 100644 --- a/aif360/algorithms/__init__.py +++ b/aif360/algorithms/__init__.py @@ -1,2 +1 @@ from aif360.algorithms.transformer import Transformer, addmetadata -from aif360.algorithms.intersectional_fairness import IntersectionalFairness diff --git a/docs/source/modules/algorithms.rst b/docs/source/modules/algorithms.rst index eaa102d3..51f20cd6 100644 --- a/docs/source/modules/algorithms.rst +++ b/docs/source/modules/algorithms.rst @@ -73,4 +73,4 @@ Algorithms :template: class.rst algorithms.Transformer - algorithms.IntersectionalFairness + algorithms.intersectional_fairness.IntersectionalFairness diff --git a/setup.py b/setup.py index 25fd4454..e26b5259 100644 --- a/setup.py +++ b/setup.py @@ -57,7 +57,7 @@ 'numpy>=1.16', 'scipy>=1.2.0', 'pandas>=0.24.0', - 'scikit-learn>=1.0', + 'scikit-learn>=1.0,<1.6', 'matplotlib', ], extras_require=extras,