Fix tests

wearepal · Dec 5, 2024 · a5827f0 · a5827f0
1 parent 51c235b
commit a5827f0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,7 +40,7 @@ repository = "https://github.com/wearepal/EthicML"
 models = [
     "GitPython<4.0.0,>=3.1.20",
     "cloudpickle<3.0.0,>=2.0.0",
-    "fairlearn==0.8.0",
+    "fairlearn>=0.11.0",
     "gitdb2==4.0.2",
     "pdm<3.0.0,>=2.4.0",
     "scipy<2.0.0,>=1.7.2",

diff --git a/tests/models_test/preprocess_test/models_preprocessing_test.py b/tests/models_test/preprocess_test/models_preprocessing_test.py
@@ -58,7 +58,7 @@ class PreprocessTest(NamedTuple):
         name="VFAE",
         num_pos=47,
     ),
-    PreprocessTest(model=Zemel(dir=TMPDIR), name="Zemel", num_pos=51),
+    PreprocessTest(model=Zemel(dir=TMPDIR), name="Zemel", num_pos=52),
     PreprocessTest(model=Beutel(dir=TMPDIR, validation_pcnt=0.25), name="Beutel dp", num_pos=51),
     PreprocessTest(
         model=Beutel(dir=TMPDIR, epochs=5, fairness=FairnessType.eq_opp, validation_pcnt=0.25),

diff --git a/uv.lock b/uv.lock