chore: add default import from domain.py to __init__

ruflab · Oct 23, 2024 · fb5f2fa · fb5f2fa
1 parent 946ad9d
commit fb5f2fa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/simple_shapes_dataset/__init__.py b/simple_shapes_dataset/__init__.py
@@ -11,6 +11,8 @@
     Attribute,
     AttributesAdditionalArgs,
     Choice,
+    DataDomain,
+    DomainDesc,
     DomainModelVariantType,
     DomainType,
     PretrainedVisualAdditionalArgs,
@@ -52,6 +54,8 @@
     "Attribute",
     "AttributesAdditionalArgs",
     "Choice",
+    "DataDomain",
+    "DomainDesc",
     "DomainModelVariantType",
     "DomainType",
     "PretrainedVisualAdditionalArgs",