Skip to content

Commit

Permalink
minor changes in the class Validator
Browse files Browse the repository at this point in the history
  • Loading branch information
Hanna Imshenetska authored and Hanna Imshenetska committed Nov 14, 2024
1 parent 0f5d5c8 commit 490cec8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/syngen/ml/config/validation.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from slugify import slugify
from loguru import logger
from syngen.ml.data_loaders import MetadataLoader, DataLoader
from syngen.ml.validation_schema import ValidationSchema
from syngen.ml.validation_schema import ValidationSchema, INFER_REPORT_TYPES


@dataclass
Expand Down Expand Up @@ -69,7 +69,7 @@ def _check_conditions(self, metadata: Dict) -> bool:
self.type_of_process == "infer"
or (
self.type_of_process == "train" and
any([item in ["accuracy", "metrics_only"] for item in reports])
any([item in INFER_REPORT_TYPES for item in reports])
)
)

Expand Down

0 comments on commit 490cec8

Please sign in to comment.