diff --git a/autotm/fitness/tasks.py b/autotm/fitness/tasks.py index 4869797..af73416 100644 --- 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 index 7a1c533..81511bd 100644 --- 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 index f8d451b..650f5e1 100644 --- 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 ",