Fix black formatting

Author: mexanick

src/dirac_cwl_proto/metadata/core.py

@@ -65,7 +65,9 @@ class DataCatalogInterface(ABC):
     """Abstract interface for data catalog operations."""
 
     @abstractmethod
-    def get_input_query(self, input_name: str, **kwargs: Any) -> Union[Path, List[Path], None]:
+    def get_input_query(
+        self, input_name: str, **kwargs: Any
+    ) -> Union[Path, List[Path], None]:
         """Generate input data query.
 
         Parameters
@@ -157,7 +159,9 @@ def post_process(self, job_path: Path) -> bool:
         """Default post-processing: always succeed."""
         return True
 
-    def get_input_query(self, input_name: str, **kwargs: Any) -> Union[Path, List[Path], None]:
+    def get_input_query(
+        self, input_name: str, **kwargs: Any
+    ) -> Union[Path, List[Path], None]:
         """Default input query: return None."""
         return None
 
@@ -196,7 +200,9 @@ class DataManager(BaseModel):
         },
     )
 
-    metadata_class: str = Field(default="User", description="Registry key for the metadata implementation class")
+    metadata_class: str = Field(
+        default="User", description="Registry key for the metadata implementation class"
+    )
 
     vo: Optional[str] = Field(
         default=None,
@@ -206,7 +212,9 @@ class DataManager(BaseModel):
     version: Optional[str] = Field(default=None, description="Metadata model version")
 
     # Enhanced fields for submission functionality
-    query_params: Dict[str, Any] = Field(default_factory=dict, description="Additional parameters for metadata plugins")
+    query_params: Dict[str, Any] = Field(
+        default_factory=dict, description="Additional parameters for metadata plugins"
+    )
 
     def model_copy_with_merge(
         self,
@@ -221,7 +229,11 @@ def model_copy_with_merge(
         # Handle nested dictionary merging for vo-specific fields
         merged_update = {}
         for key, value in update.items():
-            if hasattr(self, key) and isinstance(getattr(self, key), dict) and isinstance(value, dict):
+            if (
+                hasattr(self, key)
+                and isinstance(getattr(self, key), dict)
+                and isinstance(value, dict)
+            ):
                 existing_value = getattr(self, key).copy()
                 existing_value.update(value)
                 merged_update[key] = existing_value
@@ -285,7 +297,9 @@ def _dash_to_snake(s: str) -> str:
             return s.replace("-", "_")
 
         if submitted is None:
-            descriptor = DataManager(metadata_class=self.metadata_class, **self.query_params)
+            descriptor = DataManager(
+                metadata_class=self.metadata_class, **self.query_params
+            )
             return get_registry().instantiate_plugin(descriptor)
 
         # Build inputs from task defaults and parameter overrides
@@ -364,11 +378,17 @@ class JobExecutor(BaseModel):
 
     model_config = ConfigDict(extra="forbid", validate_assignment=True)
 
-    platform: Optional[str] = Field(default=None, description="Target platform (e.g., 'DIRAC', 'DIRACX')")
+    platform: Optional[str] = Field(
+        default=None, description="Target platform (e.g., 'DIRAC', 'DIRACX')"
+    )
 
-    priority: int = Field(default=10, description="Job priority (higher values = higher priority)")
+    priority: int = Field(
+        default=10, description="Job priority (higher values = higher priority)"
+    )
 
-    sites: Optional[List[str]] = Field(default=None, description="Candidate execution sites")
+    sites: Optional[List[str]] = Field(
+        default=None, description="Candidate execution sites"
+    )
 
     @classmethod
     def from_cwl_hints(cls: type[T], cwl_object: Any) -> T:

test/test_metadata_core.py

@@ -61,7 +61,9 @@ def test_concrete_implementation(self):
         """Test that concrete implementations work correctly."""
 
         class ConcreteCatalog(DataCatalogInterface):
-            def get_input_query(self, input_name: str, **kwargs: Any) -> Union[Path, List[Path], None]:
+            def get_input_query(
+                self, input_name: str, **kwargs: Any
+            ) -> Union[Path, List[Path], None]:
                 return Path(f"/data/{input_name}")
 
             def get_output_query(self, output_name: str) -> Optional[Path]:
@@ -225,7 +227,9 @@ def test_model_copy_with_merge(self):
         descriptor = DataManager(metadata_class="LHCbSimulation", vo="lhcb")
 
         # Test basic update
-        updated = descriptor.model_copy_with_merge(update={"metadata_class": "NewClass", "new_field": "value"})
+        updated = descriptor.model_copy_with_merge(
+            update={"metadata_class": "NewClass", "new_field": "value"}
+        )
 
         assert updated.metadata_class == "NewClass"
         assert updated.vo == "lhcb"
@@ -252,7 +256,9 @@ def test_creation(self):
 
     def test_creation_with_values(self):
         """Test JobExecutor creation with values."""
-        descriptor = JobExecutor(platform="DIRAC", priority=5, sites=["LCG.CERN.ch", "LCG.IN2P3.fr"])
+        descriptor = JobExecutor(
+            platform="DIRAC", priority=5, sites=["LCG.CERN.ch", "LCG.IN2P3.fr"]
+        )
         assert descriptor.platform == "DIRAC"
         assert descriptor.priority == 5
         assert descriptor.sites == ["LCG.CERN.ch", "LCG.IN2P3.fr"]
@@ -277,7 +283,9 @@ def test_from_cwl_hints(self, mocker):
 
     def test_serialization(self):
         """Test JobExecutor serialization."""
-        descriptor = JobExecutor(platform="DIRAC", priority=7, sites=["LCG.CERN.ch", "LCG.IN2P3.fr"])
+        descriptor = JobExecutor(
+            platform="DIRAC", priority=7, sites=["LCG.CERN.ch", "LCG.IN2P3.fr"]
+        )
 
         # Test model serialization
         data = descriptor.model_dump()
@@ -292,7 +300,9 @@ class TestTransformationDataManager:
 
     def test_creation(self):
         """Test TransformationDataManager creation."""
-        descriptor = TransformationDataManager(metadata_class="QueryBased", group_size={"input_data": 100})
+        descriptor = TransformationDataManager(
+            metadata_class="QueryBased", group_size={"input_data": 100}
+        )
         assert descriptor.metadata_class == "QueryBased"
         assert descriptor.group_size == {"input_data": 100}
 
@@ -314,7 +324,9 @@ def test_inheritance(self):
     def test_validation(self):
         """Test group_size validation."""
         # Valid group_size
-        descriptor = TransformationDataManager(metadata_class="User", group_size={"files": 10})
+        descriptor = TransformationDataManager(
+            metadata_class="User", group_size={"files": 10}
+        )
         assert descriptor.group_size == {"files": 10}
 
         # Test with no group_size