UncountablePythonSDK 0.0.41__py3-none-any.whl → 0.0.42__py3-none-any.whl

uncountable/types/api/recipes/set_recipe_tags.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from ... import base_t
 from ... import identifier_t
@@ -38,7 +38,7 @@ class RecipeTagUpdateType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeTagUpdateBase:
     type: RecipeTagUpdateType
 
@@ -47,7 +47,7 @@ class RecipeTagUpdateBase:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeTagAppend(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.APPEND] = RecipeTagUpdateType.APPEND
     recipe_tag_ids: list[base_t.ObjectId]
@@ -57,7 +57,7 @@ class RecipeTagAppend(RecipeTagUpdateBase):
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeTagRemove(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.REMOVE] = RecipeTagUpdateType.REMOVE
     recipe_tag_ids: list[base_t.ObjectId]
@@ -67,7 +67,7 @@ class RecipeTagRemove(RecipeTagUpdateBase):
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeTagOverride(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.OVERRIDE] = RecipeTagUpdateType.OVERRIDE
     recipe_tag_ids: list[base_t.ObjectId]
@@ -78,14 +78,14 @@ RecipeTagUpdate = typing.Union[RecipeTagAppend, RecipeTagRemove, RecipeTagOverri
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     recipe_key: identifier_t.IdentifierKey
     recipe_tag_update: RecipeTagUpdate
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/unarchive_recipes.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -22,13 +22,13 @@ ENDPOINT_PATH = "api/external/recipes/unarchive"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     recipes: list[identifier_t.IdentifierKey]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/unlock_recipes.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -29,7 +29,7 @@ class RecipeUnlockType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     type: RecipeUnlockType = RecipeUnlockType.STANDARD
     recipes: list[identifier_t.IdentifierKey]
@@ -37,7 +37,7 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/triggers/run_trigger.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from ... import entity_t
 from ... import response_t
 
@@ -23,14 +23,14 @@ ENDPOINT_PATH = "api/external/triggers/run_trigger"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     trigger_ref_name: str
     entity: typing.Optional[entity_t.Entity] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data(response_t.Response):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/uploader/__init__.py

@@ -0,0 +1 @@
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/uploader/invoke_uploader.py

@@ -0,0 +1,38 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from ... import async_batch_t
+from ... import base_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/uploader/invoke_uploader"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    file_ids: list[base_t.ObjectId]
+    uploader_key: identifier_t.IdentifierKey
+    material_family_keys: list[identifier_t.IdentifierKey]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class Data(async_batch_t.AsyncBatchActionReturn):
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch_processor.py

@@ -14,6 +14,7 @@ from uncountable.types import base_t
 import uncountable.types.api.recipes.create_recipe as create_recipe_t
 import uncountable.types.api.recipes.edit_recipe_inputs as edit_recipe_inputs_t
 from uncountable.types import identifier_t
+import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
 from uncountable.types import recipe_identifiers_t
 from uncountable.types import recipe_metadata_t
 from uncountable.types import recipe_workflow_steps_t
@@ -156,6 +157,41 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def invoke_uploader(
+        self,
+        *,
+        file_ids: list[base_t.ObjectId],
+        uploader_key: identifier_t.IdentifierKey,
+        material_family_keys: list[identifier_t.IdentifierKey],
+        depends_on: typing.Optional[list[str]] = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Runs files through an uploader.
+
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = invoke_uploader_t.Arguments(
+            file_ids=file_ids,
+            uploader_key=uploader_key,
+            material_family_keys=material_family_keys,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.INVOKE_UPLOADER,
+            data=json_data,
+            depends_on=depends_on,
+            batch_reference=batch_reference,
+        )
+
+        self._enqueue(req)
+
+        return async_batch_t.QueuedAsyncBatchRequest(
+            path=req.path,
+            batch_reference=req.batch_reference,
+        )
+
     def set_recipe_metadata(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
 
@@ -30,13 +30,15 @@ class AsyncBatchRequestPath(StrEnum):
     LOCK_RECIPES = "recipes/lock_recipes"
     UNLOCK_RECIPES = "recipes/unlock_recipes"
     ASSOCIATE_EQUIPMENT_INPUT = "equipment/associate_equipment_input"
+    INVOKE_UPLOADER = "uploader/invoke_uploader"
+    ASSOCIATE_RECIPE_AS_INPUT = "recipes/associate_recipe_as_input"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     unconverted_values={"data"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class AsyncBatchRequest:
     path: AsyncBatchRequestPath
     data: base_t.JsonValue
@@ -45,14 +47,14 @@ class AsyncBatchRequest:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class AsyncBatchActionReturn:
     modification_made: bool
     result_id: typing.Optional[base_t.ObjectId] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
 class QueuedAsyncBatchRequest:
     path: AsyncBatchRequestPath
     batch_reference: str

uncountable/types/calculations_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Calculation:
     id: base_t.ObjectId
     name: str

uncountable/types/chemical_structure_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 
 __all__: list[str] = [
@@ -19,7 +19,7 @@ __all__: list[str] = [
 @serial_class(
     to_string_values={"molecular_weight"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class ChemicalStructure:
     ketcher_file: typing.Optional[str] = None
     gross_formula: typing.Optional[str] = None

uncountable/types/client_base.py

@@ -47,6 +47,7 @@ import uncountable.types.api.recipes.get_recipe_output_metadata as get_recipe_ou
 import uncountable.types.api.recipes.get_recipes_data as get_recipes_data_t
 from uncountable.types import id_source_t
 from uncountable.types import identifier_t
+import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
 import uncountable.types.api.entity.list_entities as list_entities_t
 import uncountable.types.api.id_source.list_id_source as list_id_source_t
 import uncountable.types.api.entity.lock_entity as lock_entity_t
@@ -81,12 +82,12 @@ import uncountable.types.api.recipes.unlock_recipes as unlock_recipes_t
 import uncountable.types.api.material_families.update_entity_material_families as update_entity_material_families_t
 import uncountable.types.api.field_options.upsert_field_options as upsert_field_options_t
 from abc import ABC, abstractmethod
-from dataclasses import dataclass
+import dataclasses
 
 DT = typing.TypeVar("DT")
 
 
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class APIRequest:
     method: str
     endpoint: str
@@ -850,6 +851,28 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=get_recipes_data_t.Data)
 
+    def invoke_uploader(
+        self,
+        *,
+        file_ids: list[base_t.ObjectId],
+        uploader_key: identifier_t.IdentifierKey,
+        material_family_keys: list[identifier_t.IdentifierKey],
+    ) -> invoke_uploader_t.Data:
+        """Runs files through an uploader.
+
+        """
+        args = invoke_uploader_t.Arguments(
+            file_ids=file_ids,
+            uploader_key=uploader_key,
+            material_family_keys=material_family_keys,
+        )
+        api_request = APIRequest(
+            method=invoke_uploader_t.ENDPOINT_METHOD,
+            endpoint=invoke_uploader_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=invoke_uploader_t.Data)
+
     def list_entities(
         self,
         *,

uncountable/types/curves_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
 
@@ -26,7 +26,7 @@ class CurveAxisType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Curve:
     id: base_t.ObjectId
     x_type: typing.Union[typing.Literal[CurveAxisType.NUMERIC], typing.Literal[CurveAxisType.TIMESTAMP]]
@@ -39,7 +39,7 @@ class Curve:
 @serial_class(
     to_string_values={"quantity"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class CurveValue:
     index: int
     quantity: typing.Optional[Decimal] = None

uncountable/types/entity_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_string_enum
 from . import base_t
 
@@ -267,7 +267,7 @@ class EntityType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Entity:
     id: base_t.ObjectId
     type: EntityType

uncountable/types/experiment_groups_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleExperimentGroup:
     experiment_group_id: base_t.ObjectId
     name: str

uncountable/types/field_values_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
 
@@ -23,7 +23,7 @@ __all__: list[str] = [
 @serial_class(
     unconverted_values={"value"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class FieldRefNameValue:
     field_ref_name: str
     value: base_t.JsonValue
@@ -34,7 +34,7 @@ class FieldRefNameValue:
 @serial_class(
     unconverted_values={"value"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class FieldRefIdNameValue:
     field_id: base_t.ObjectId
     field_ref_name: str
@@ -46,7 +46,7 @@ class FieldRefIdNameValue:
 @serial_class(
     unconverted_values={"value"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class ArgumentValueRefName:
     field_ref_name: str
     value: base_t.JsonValue
@@ -57,7 +57,7 @@ class ArgumentValueRefName:
 @serial_class(
     unconverted_values={"value"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class ArgumentValueId:
     field_id: base_t.ObjectId
     value: base_t.JsonValue

uncountable/types/fields_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Field:
     id: base_t.ObjectId
     name: str

uncountable/types/generic_upload.py

@@ -0,0 +1,9 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .generic_upload_t import GenericRemoteDirectoryScope as GenericRemoteDirectoryScope
+from .generic_upload_t import GenericUploadStrategy as GenericUploadStrategy
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/generic_upload_t.py

@@ -0,0 +1,41 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from . import identifier_t
+
+__all__: list[str] = [
+    "GenericRemoteDirectoryScope",
+    "GenericUploadStrategy",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class GenericRemoteDirectoryScope:
+    src_path: str
+    success_archive_path: str
+    failure_archive_path: str
+    recursive: bool
+    prepend_date_on_archive: bool
+    valid_file_extensions: list[str]
+    filename_regex: typing.Optional[str] = None
+    max_files: typing.Optional[int] = None
+    detection_keyword: typing.Optional[str] = None
+    delimiter: typing.Optional[str] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class GenericUploadStrategy:
+    uploader_key: identifier_t.IdentifierKey
+    material_family_keys: list[identifier_t.IdentifierKey]
+    parse_files_individually: bool
+    skip_moving_files: bool = False
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/id_source_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import entity_t
 
 __all__: list[str] = [
@@ -20,25 +20,25 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class IdSourceSpecBase:
     pass
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class IdSourceSpecEntity(IdSourceSpecBase):
     entity_type: entity_t.EntityType
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class IdSourceSpecCustomEntity(IdSourceSpecBase):
     definition_ref_name: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class IdSourceSpecFieldOptions(IdSourceSpecBase):
     set_ref_name: str
     subset_ref_name: typing.Optional[str] = None

uncountable/types/identifier_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
 
@@ -23,7 +23,7 @@ __all__: list[str] = [
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
 class IdentifierKeyId:
     type: typing.Literal["id"] = "id"
     id: base_t.ObjectId
@@ -33,7 +33,7 @@ class IdentifierKeyId:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
 class IdentifierKeyRefName:
     type: typing.Literal["ref_name"] = "ref_name"
     ref_name: str
@@ -43,7 +43,7 @@ class IdentifierKeyRefName:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
 class IdentifierKeyBatchReference:
     type: typing.Literal["batch_reference"] = "batch_reference"
     reference: str

uncountable/types/input_attributes_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
 
@@ -21,7 +21,7 @@ __all__: list[str] = [
     unconverted_values={"quantity_json"},
     to_string_values={"quantity_dec"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class InputAttributeValue:
     attribute_id: base_t.ObjectId
     quantity_dec: typing.Optional[Decimal] = None

uncountable/types/inputs_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_string_enum
 from . import base_t
 
@@ -51,7 +51,7 @@ class IngredientType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleInput:
     input_id: base_t.ObjectId
     quantity_type: IngredientQuantityType