clean up

vanderschaarlab · Oct 1, 2024 · 6043e1b · 6043e1b
1 parent d536d2e
commit 6043e1b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -47,7 +47,7 @@ install_requires =
     tenacity
     tqdm
     loguru
-    pydantic
+    pydantic>=2.0
     cloudpickle
     scipy
     xgboost<3.0.0

diff --git a/tests/plugins/core/test_distribution.py b/tests/plugins/core/test_distribution.py
@@ -49,7 +49,6 @@ def test_categorical() -> None:
     assert param.includes(param_other)
     assert param_other.includes(param)
 
-    # Instead of asserting marginal_distribution is None, assert it's correctly initialized
     expected_marginal = pd.Series(
         [0.25, 0.25, 0.25, 0.25], index=["1", "2", "55", "sdfsf"]
     )
@@ -195,7 +194,6 @@ def test_float_from_data() -> None:
     assert param.includes(param_other)
     assert not param_other.includes(param)
 
-    # This assertion should now pass
     assert param.marginal_distribution is not None