From b6d07b28feac0c3ba4dfa2b5b89e515c0820a33f Mon Sep 17 00:00:00 2001 From: Saskia de Vries Date: Sat, 16 Nov 2024 21:05:24 -0800 Subject: [PATCH] linting --- src/aind_data_schema/core/model.py | 2 +- tests/test_model.py | 13 +++++++------ 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/src/aind_data_schema/core/model.py b/src/aind_data_schema/core/model.py index 3ab05eb68..af696f66d 100644 --- a/src/aind_data_schema/core/model.py +++ b/src/aind_data_schema/core/model.py @@ -32,7 +32,7 @@ class ModelEvaluation(AindModel): """Description of model evaluation""" data: Optional[str] = Field(default=None, title="Path to evaluation data") - data_description: Optional[str] = Field(default=None, title="Description of evaluation data") + data_description: Optional[str] = Field(default=None, title="Description of evaluation data") date: AwareDatetimeWithDefault = Field(..., title="Date") performance: List[PerformanceMetric] = Field(..., title="Evaluation performance") notes: Optional[str] = Field(default=None, title="Notes") diff --git a/tests/test_model.py b/tests/test_model.py index 999fd981e..f1af9ece5 100644 --- a/tests/test_model.py +++ b/tests/test_model.py @@ -7,12 +7,12 @@ from aind_data_schema_models.modalities import Modality from aind_data_schema_models.organizations import Organization -from aind_data_schema_models.process_names import ProcessName from aind_data_schema_models.system_architecture import ModelBackbone from aind_data_schema.core.model import Model, ModelArchitecture, ModelEvaluation, ModelTraining, PerformanceMetric from aind_data_schema.components.devices import Software + class ModelTests(unittest.TestCase): """tests for model""" @@ -36,9 +36,9 @@ def test_constructors(self): parameters={ "downsample": 1, "input_shape": [ - 14, - 14, - 26 + 14, + 14, + 26 ], "learning_rate": 0.0001, "train_test_split": 0.8, @@ -86,11 +86,12 @@ def test_constructors(self): ) ] ) - + Model.model_validate_json(m.model_dump_json()) self.assertIsNotNone(m) if __name__ == "__main__": - unittest.main() \ No newline at end of file + unittest.main() + \ No newline at end of file