lint

Author: andreatgretel

src/data_designer/interface/results.py

@@ -69,7 +69,9 @@ def load_processor_dataset(self, processor_name: str) -> pd.DataFrame:
             A pandas DataFrame containing the dataset generated by the processor.
         """
         try:
-            dataset = self.artifact_storage.read_parquet_files(self.artifact_storage.processors_outputs_path / processor_name)
+            dataset = self.artifact_storage.read_parquet_files(
+                self.artifact_storage.processors_outputs_path / processor_name
+            )
         except Exception as e:
             raise ArtifactStorageError(f"Failed to load dataset for processor {processor_name}: {e}")
 

tests/config/utils/test_validation.py

@@ -286,11 +286,17 @@ def test_validate_ancillary_dataset_processor():
     assert len(violations) == 2
     assert violations[0].type == ViolationType.INVALID_REFERENCE
     assert violations[0].column is None
-    assert violations[0].message == "Ancillary dataset processor attempts to reference columns 'invalid_reference' in the template for 'text', but the columns are not defined in the dataset."
+    assert (
+        violations[0].message
+        == "Ancillary dataset processor attempts to reference columns 'invalid_reference' in the template for 'text', but the columns are not defined in the dataset."
+    )
     assert violations[0].level == ViolationLevel.ERROR
     assert violations[1].type == ViolationType.INVALID_TEMPLATE
     assert violations[1].column is None
-    assert violations[1].message == "Ancillary dataset processor ancillary_dataset_processor_invalid_template template is not a valid JSON object."
+    assert (
+        violations[1].message
+        == "Ancillary dataset processor ancillary_dataset_processor_invalid_template template is not a valid JSON object."
+    )
     assert violations[1].level == ViolationLevel.ERROR