Bug fixes for pure python mode

seifashraf1 · Jan 19, 2023 · c5a5515 · c5a5515
1 parent 2508448
commit c5a5515
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/runtime/ltypes/ltypes.py b/src/runtime/ltypes/ltypes.py
@@ -24,7 +24,7 @@ def __call__(self, arg):
         return arg
 
 def is_dataclass_wrapper(obj):
-    if issubclass(obj, PackedDataClass):
+    if isclass(obj) and issubclass(obj, PackedDataClass):
         return is_dataclass(obj.class_to_pack)
     return is_dataclass(obj)
 
@@ -147,14 +147,14 @@ def packed(*args, aligned=None):
     if len(args) == 1:
         if not is_dataclass_wrapper(args[0]):
             raise TypeError("packed can only be applied over a dataclass.")
-        class PackedDataClassLocal(PackedDataClass):
+        class PackedDataClassLocal(args[0], PackedDataClass):
             class_to_pack = args[0]
         return PackedDataClassLocal
 
     def _packed(f):
         if not is_dataclass_wrapper(f):
             raise TypeError("packed can only be applied over a dataclass.")
-        class PackedDataClassLocal(PackedDataClass):
+        class PackedDataClassLocal(f, PackedDataClass):
             class_to_pack = f
         return PackedDataClassLocal
     return _packed