Properly output frozenset object in pycdas

mrtruongleo · Feb 14, 2023 · 80e835e · 80e835e
1 parent f00921d
commit 80e835e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pycdas.cpp b/pycdas.cpp
@@ -211,6 +211,14 @@ void output_object(PycRef<PycObject> obj, PycModule* mod, int indent,
             iputs(indent, "}\n");
         }
         break;
+    case PycObject::TYPE_FROZENSET:
+        {
+            iputs(indent, "frozenset({\n");
+            for (const auto& val : obj.cast<PycSet>()->values())
+                output_object(val, mod, indent + 1, flags);
+            iputs(indent, "})\n");
+        }
+        break;
     case PycObject::TYPE_NONE:
         iputs(indent, "None\n");
         break;