ruff it

Author: casenave

src/plaid/infos.py

@@ -11,7 +11,14 @@
 
 logger = logging.getLogger(__name__)
 
-class DataProduction(BaseModel, revalidate_instances="always", str_strip_whitespace=True, validate_assignment=True, extra="forbid"):
+
+class DataProduction(
+    BaseModel,
+    revalidate_instances="always",
+    str_strip_whitespace=True,
+    validate_assignment=True,
+    extra="forbid",
+):
     """Dataset production context metadata."""
 
     owner: str | None = None
@@ -36,7 +43,13 @@ class DataProduction(BaseModel, revalidate_instances="always", str_strip_whitesp
 )
 
 
-class Infos(BaseModel, revalidate_instances="always", str_strip_whitespace=True, validate_assignment=True, extra="forbid"):
+class Infos(
+    BaseModel,
+    revalidate_instances="always",
+    str_strip_whitespace=True,
+    validate_assignment=True,
+    extra="forbid",
+):
     """Structured representation of a PLAID dataset ``infos`` payload."""
 
     # model_config = _PD_CONFIG

src/plaid/problem_definition.py

@@ -26,7 +26,13 @@ def _normalize_list(v):
     return sorted(map(str, v))
 
 
-class ProblemDefinition(BaseModel, revalidate_instances="always", str_strip_whitespace=True, validate_assignment=True, extra="forbid"):
+class ProblemDefinition(
+    BaseModel,
+    revalidate_instances="always",
+    str_strip_whitespace=True,
+    validate_assignment=True,
+    extra="forbid",
+):
     """Defines the input and output features for a machine learning problem."""
 
     # model_config = ConfigDict(

tests/containers/test_sample.py

@@ -1349,7 +1349,7 @@ def test_save(self, sample_with_tree_and_scalar, tmp_path):
             save_dir, overwrite=True, memory_safe=True
         )
 
-    def test_load_from_dir(self, sample_with_tree_and_scalar, tmp_path):
+    def test_load_from_dir_2(self, sample_with_tree_and_scalar, tmp_path):
         save_dir = tmp_path / "test_dir"
         sample_with_tree_and_scalar.save_to_dir(save_dir)
         new_sample = Sample.load_from_dir(save_dir)