used black

AllenNeuralDynamics · Dec 12, 2023 · 27ecdc3 · 27ecdc3
1 parent e62db21
commit 27ecdc3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 10 deletions.
diff --git a/src/aind_metadata_mapper/bergamo/session.py b/src/aind_metadata_mapper/bergamo/session.py
@@ -19,10 +19,7 @@
     Session,
     Stream,
 )
-from aind_data_schema.models.devices import (
-    Detector,
-    Laser,
-)
+from aind_data_schema.models.devices import Detector, Laser
 from aind_data_schema.models.stimulus import (
     PhotoStimulation,
     PhotoStimulationGroup,

diff --git a/src/aind_metadata_mapper/core.py b/src/aind_metadata_mapper/core.py
@@ -84,15 +84,14 @@ def _run_validation_check(model_instance: AindCoreModel) -> None:
           Contents from the service response.
         """
         try:
-            model_instance.model_validate(
-                model_instance.__dict__
-            )
+            model_instance.model_validate(model_instance.__dict__)
             logging.debug("No validation errors detected.")
         except ValidationError:
             logging.warning(
                 "Validation errors were found. This may be due to "
                 "mismatched versions or data not found in the "
-                "databases.", exc_info=True,
+                "databases.",
+                exc_info=True,
             )
 
     def run_job(self) -> None:

diff --git a/tests/test_subject.py b/tests/test_subject.py
@@ -101,7 +101,8 @@ def test_successful_response(
         self.subject_etl.run_job()
         mock_write.assert_called_once_with(output_directory=Path("tests"))
         mock_log_warn.assert_called_once_with(
-            self.validation_warning_message, exc_info=True)
+            self.validation_warning_message, exc_info=True
+        )
 
     @patch(
         "aind_metadata_service.client.AindMetadataServiceClient.get_subject"
@@ -226,7 +227,7 @@ def test_mocked_validate_model(
         mocked_response = self.invalid_response
         mocked_response.status_code = StatusCodes.VALID_DATA
         mock_api_get.return_value = mocked_response
-        mock_validate.return_value = (None, None)
+        # mock_validate.return_value = (None, None)
         self.subject_etl.run_job()
         mock_write.assert_called_once_with(output_directory=Path("tests"))