fix metric assert bug and update reqs

WecoAI · Apr 26, 2024 · 5807267 · 5807267
1 parent 6fafd97
commit 5807267
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/aide/utils/metric.py b/aide/utils/metric.py
@@ -38,7 +38,6 @@ def __gt__(self, other) -> bool:
         return comp if self.maximize else not comp  # type: ignore
 
     def __eq__(self, other: Any) -> bool:
-        assert type(self) is type(other)
         return self.value == other.value
 
     def __repr__(self) -> str:

diff --git a/requirements.txt b/requirements.txt
@@ -60,7 +60,7 @@ hyperopt
 bayesian-optimization
 imagecodecs
 hmmlearn
-bayespy
+bayespy==0.5.1
 sklearn-pandas
 tensorpack
 sentencepiece

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name="aideml",
-    version="0.1.1",
+    version="0.1.2",
     author="Weco AI",
     author_email="[email protected]",
     description="Autonomous AI for Data Science and Machine Learning",