flake8 compliant

src/model_textual/classes.py

@@ -14,13 +14,13 @@ class ModelOutput(BaseModel):
     def list_fields(cls) -> List[str]:
         """List options that are stored as attributes."""
         return list(cls.model_fields.keys())
-    
+
     @classmethod
     def from_random(cls):
         """Instantiate with random numbers."""
         kwargs = {field: random.random() for field in cls.list_fields()}
         return cls(**kwargs)
-    
+
     @classmethod
     def from_choice(cls, option: str):
         """Instantiate from choice."""
@@ -28,7 +28,8 @@ class ModelOutput(BaseModel):
             raise ValidationError()
         assert isinstance(option, str)
         allowed_options_list = cls.list_fields()
-        assert option in allowed_options_list, f"{option} is not among allowed fields {allowed_options_list}"
+        assert option in allowed_options_list, \
+            f"{option} is not among allowed fields {allowed_options_list}"
         kwargs = {field: 0 for field in cls.list_fields()}
         kwargs[option] = 1
         return cls(**kwargs)
@@ -36,15 +37,19 @@ class ModelOutput(BaseModel):
     def __repr__(self) -> str:
         model_dict = self.model_dump()
         model_repr_str = f"{self.classname()}("
-        model_repr_str += ", ".join([f"{field}={value:.3f}" for field, value in model_dict.items()])
+        model_repr_str += ", ".join([
+            f"{field}={value:.3f}"
+            for field, value
+            in model_dict.items()
+        ])
         model_repr_str += ")"
         return model_repr_str
-    
+
     def highest_score_field(self) -> str:
         """Return name of field with highest score."""
         model_dict = self.model_dump()
         return max(model_dict, key=lambda k: model_dict[k])
-    
+
     def highest_score_value(self) -> float:
         """Return value of field with highest score."""
         model_dict = self.model_dump()
@@ -84,4 +89,4 @@ if __name__ == '__main__':
     m = SalienceModelOutput.from_random()
     print(repr(m))
     print(m.highest_score_field())
-    print(m.highest_score_value())
+    print(m.highest_score_value())