From 5807267adeb9b4926c1525dc91d5151e9603cd26 Mon Sep 17 00:00:00 2001 From: Dominik Schmidt Date: Fri, 26 Apr 2024 12:50:31 +0100 Subject: [PATCH] fix metric assert bug and update reqs --- aide/utils/metric.py | 1 - requirements.txt | 2 +- setup.py | 2 +- 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/aide/utils/metric.py b/aide/utils/metric.py index a881215..d60c80d 100644 --- 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 index 3f065c0..5248639 100644 --- 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 index 318950d..7e7efc7 100644 --- 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="contact@weco.ai", description="Autonomous AI for Data Science and Machine Learning",