wrap a fields in a sequence if it is a simple string fixing deschler#766

hapytex · Nov 13, 2024 · 5d4a619 · 5d4a619
1 parent 294302c
commit 5d4a619
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modeltranslation/translator.py b/modeltranslation/translator.py
@@ -77,7 +77,11 @@ class FieldsAggregationMetaClass(type):
     fields: Sequence[str]
 
     def __new__(cls, name: str, bases: tuple[type, ...], attrs: dict[str, Any]) -> type:
-        attrs["fields"] = set(attrs.get("fields", ()))
+        fields = attrs.get("fields", ())
+        if isinstance(fields, str):
+            attrs['fields'] = {fields}
+        else:
+            attrs["fields"] = set(fields)
         for base in bases:
             if isinstance(base, FieldsAggregationMetaClass):
                 attrs["fields"].update(base.fields)