kw_only support

Author: I8dNLo

fastembed/common/model_description.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass, field, InitVar
 from typing import Optional, List, Dict
 
 
@@ -28,12 +28,52 @@ class ModelDescription:
     tasks: Dict[str, int] = field(default_factory=dict)
 
 
-@dataclass(frozen=True, kw_only=True)
+# @dataclass(frozen=True, kw_only=True)
+# class SparseModelDescription(ModelDescription):
+#     vocab_size: int
+#     requires_idf: Optional[bool] = None
+#     # For sparse models, override dim to always be None.
+#     dim: Optional[int] = None
+
+
+@dataclass(frozen=True)
 class SparseModelDescription(ModelDescription):
-    vocab_size: int
-    requires_idf: Optional[bool] = None
-    # For sparse models, override dim to always be None.
-    dim: Optional[int] = None
+    _vocab_size: InitVar[Optional[int]] = None
+    _requires_idf: InitVar[Optional[bool]] = None
+
+    vocab_size: int = field(init=False)
+    requires_idf: Optional[bool] = field(init=False, default=None)
+    dim: Optional[int] = field(default=None, init=False)  # Always None for sparse models.
+
+    def __init__(
+        self,
+        *,
+        model: str,
+        sources: ModelSource,
+        model_file: str,
+        description: str,
+        license: str,
+        size_in_GB: float,
+        additional_files: Optional[List[str]] = None,
+        tasks: Optional[Dict[str, int]] = None,
+        vocab_size: int,
+        requires_idf: Optional[bool] = None,
+    ):
+        # Call the parent initializer with the fields it needs.
+        object.__setattr__(self, "model", model)
+        object.__setattr__(self, "sources", sources)
+        object.__setattr__(self, "model_file", model_file)
+        object.__setattr__(self, "dim", None)
+        object.__setattr__(self, "description", description)
+        object.__setattr__(self, "license", license)
+        object.__setattr__(self, "size_in_GB", size_in_GB)
+        object.__setattr__(
+            self, "additional_files", additional_files if additional_files is not None else []
+        )
+        object.__setattr__(self, "tasks", tasks if tasks is not None else {})
+        # Set new fields.
+        object.__setattr__(self, "vocab_size", vocab_size)
+        object.__setattr__(self, "requires_idf", requires_idf)
 
 
 @dataclass(frozen=True)