fix for IndividualDTO definition (Optional for MetricsScore)

aimclub · Nov 26, 2023 · e40ec5f · e40ec5f
1 parent 58d16a6
commit e40ec5f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/autotm/fitness/tasks.py b/autotm/fitness/tasks.py
@@ -16,6 +16,7 @@ def do_fitness_calculating(
     alg_args: Optional[str] = None,
     is_tmp: bool = False,
 ) -> str:
+    logger.info("Doing fitness calculating with individual %s" % individual)
     individual: IndividualDTO = IndividualDTO.parse_raw(individual)
 
     with fit_tm_of_individual(

diff --git a/autotm/schemas.py b/autotm/schemas.py
@@ -32,7 +32,7 @@ class IndividualDTO(BaseModel):
     fitness_name: str = "default"
     dataset: str = "default"
     force_dataset_settings_checkout: bool = False
-    fitness_value: MetricsScores = None
+    fitness_value: Optional[MetricsScores] = None
     exp_id: Optional[int] = None
     alg_id: Optional[str] = None
     tag: Optional[str] = None

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "autotm"
-version = "0.1.0"
+version = "0.1.1"
 description = "Automatic hyperparameters tuning for topic models (ARTM approach) using evolutionary algorithms"
 authors = [
     "Khodorchenko Maria <[email protected]>",