Merge branch 'dev' into update-version-to-post2

OpenMined · Apr 1, 2024 · 34a6a7e · 34a6a7e
2 parents 581f571 + 899a36a
commit 34a6a7e
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/packages/syft/src/syft/types/syft_metaclass.py b/packages/syft/src/syft/types/syft_metaclass.py
@@ -1,6 +1,5 @@
 # stdlib
 from typing import Any
-from typing import TypeVar
 from typing import final
 
 # third party
@@ -10,8 +9,6 @@
 # relative
 from ..serde.serializable import serializable
 
-_T = TypeVar("_T", bound=BaseModel)
-
 
 class EmptyType(type):
     def __repr__(self) -> str:
@@ -38,10 +35,12 @@ def __new__(
     ) -> type:
         cls = super().__new__(mcs, cls_name, bases, namespace, *args, **kwargs)
 
-        for field_info in cls.model_fields.values():
-            if field_info.annotation is not None and field_info.is_required():
-                field_info.annotation = field_info.annotation | EmptyType
-                field_info.default = Empty
+        if issubclass(cls, BaseModel):
+            for field_info in cls.model_fields.values():
+                if field_info.annotation is not None and field_info.is_required():
+                    field_info.annotation = field_info.annotation | EmptyType
+                    field_info.default = Empty
+
+            cls.model_rebuild(force=True)
 
-        cls.model_rebuild(force=True)
         return cls