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

uncountable/integration/secret_retrieval/retrieve_secret.py

@@ -0,0 +1,40 @@
+import os
+
+from uncountable.types.job_definition_t import ProfileMetadata
+from uncountable.types.secret_retrieval_t import (
+    SecretRetrieval,
+    SecretRetrievalAWS,
+    SecretRetrievalEnv,
+)
+
+
+class SecretRetrievalError(BaseException):
+    def __init__(
+        self, secret_retrieval: SecretRetrieval, message: str | None = None
+    ) -> None:
+        self.secret_retrieval = secret_retrieval
+        self.message = message
+
+    def __str__(self) -> str:
+        append_message = ""
+        if self.message is not None:
+            append_message = f": {self.message}"
+        return f"{self.secret_retrieval.type} secret retrieval failed{append_message}"
+
+
+def retrieve_secret(
+    secret_retrieval: SecretRetrieval, profile_metadata: ProfileMetadata
+) -> str:
+    match secret_retrieval:
+        case SecretRetrievalEnv():
+            env_name = (
+                f"UNC_{profile_metadata.name.upper()}_{secret_retrieval.env_key.upper()}"
+            )
+            secret = os.environ.get(env_name)
+            if secret is None:
+                raise SecretRetrievalError(
+                    secret_retrieval, f"environment variable {env_name} missing"
+                )
+            return secret
+        case SecretRetrievalAWS():
+            raise NotImplementedError("aws secret retrieval not yet implemented")

uncountable/integration/server.py

@@ -11,7 +11,7 @@ from apscheduler.triggers.cron import CronTrigger
 from sqlalchemy.engine.base import Engine
 
 from uncountable.integration.cron import CronJobArgs, cron_job_executor
-from uncountable.integration.types import (
+from uncountable.types.job_definition_t import (
     AuthRetrieval,
     CronJobDefinition,
     JobDefinition,

uncountable/types/__init__.py

@@ -28,6 +28,7 @@ from .api.batch import execute_batch_load_async as execute_batch_load_async_t
 from . import experiment_groups_t as experiment_groups_t
 from . import field_values_t as field_values_t
 from . import fields_t as fields_t
+from . import generic_upload_t as generic_upload_t
 from .api.recipes import get_curve as get_curve_t
 from .api.entity import get_entities_data as get_entities_data_t
 from .api.inputs import get_input_data as get_input_data_t
@@ -48,6 +49,8 @@ from . import id_source_t as id_source_t
 from . import identifier_t as identifier_t
 from . import input_attributes_t as input_attributes_t
 from . import inputs_t as inputs_t
+from .api.uploader import invoke_uploader as invoke_uploader_t
+from . import job_definition_t as job_definition_t
 from .api.entity import list_entities as list_entities_t
 from .api.id_source import list_id_source as list_id_source_t
 from .api.entity import lock_entity as lock_entity_t
@@ -71,6 +74,7 @@ from .api.entity import resolve_entity_ids as resolve_entity_ids_t
 from .api.outputs import resolve_output_conditions as resolve_output_conditions_t
 from . import response_t as response_t
 from .api.triggers import run_trigger as run_trigger_t
+from . import secret_retrieval_t as secret_retrieval_t
 from .api.permissions import set_core_permissions as set_core_permissions_t
 from .api.inputs import set_input_attribute_values as set_input_attribute_values_t
 from .api.inputs import set_input_category as set_input_category_t
@@ -119,6 +123,7 @@ __all__: list[str] = [
     "experiment_groups_t",
     "field_values_t",
     "fields_t",
+    "generic_upload_t",
     "get_curve_t",
     "get_entities_data_t",
     "get_input_data_t",
@@ -139,6 +144,8 @@ __all__: list[str] = [
     "identifier_t",
     "input_attributes_t",
     "inputs_t",
+    "invoke_uploader_t",
+    "job_definition_t",
     "list_entities_t",
     "list_id_source_t",
     "lock_entity_t",
@@ -162,6 +169,7 @@ __all__: list[str] = [
     "resolve_output_conditions_t",
     "response_t",
     "run_trigger_t",
+    "secret_retrieval_t",
     "set_core_permissions_t",
     "set_input_attribute_values_t",
     "set_input_category_t",

uncountable/types/api/batch/execute_batch.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
 
@@ -43,7 +43,7 @@ class RequestMethod(StrEnum):
 @serial_class(
     unconverted_values={"data"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class BatchRequest:
     path: str
     method: RequestMethod
@@ -51,7 +51,7 @@ class BatchRequest:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     requests: list[BatchRequest]
 
@@ -60,7 +60,7 @@ class Arguments:
 @serial_class(
     unconverted_values={"response"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class BatchResponse:
     path: str
     method: RequestMethod
@@ -69,7 +69,7 @@ class BatchResponse:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     responses: list[BatchResponse]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/batch/execute_batch_load_async.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 async_batch_t
 from ... import base_t
 
@@ -23,13 +23,13 @@ ENDPOINT_PATH = "api/external/batch/execute_batch_load_async"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     requests: list[async_batch_t.AsyncBatchRequest]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     job_id: base_t.ObjectId
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/chemical/convert_chemical_formats.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
 
@@ -25,13 +25,13 @@ ENDPOINT_PATH = "api/external/chemical/convert_chemical_formats"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class ChemicalStructureFile:
     struct_file: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     source_chemical_structures: list[ChemicalStructureFile]
 
@@ -44,7 +44,7 @@ UncountableChemicalStructure = dict[str, base_t.JsonValue]
 @serial_class(
     unconverted_values={"chemical_structures"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     chemical_structures: list[UncountableChemicalStructure]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entities.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
 from ... import entity_t
 from ... import field_values_t
@@ -25,13 +25,13 @@ ENDPOINT_PATH = "api/external/entity/external_create_entities"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class EntityToCreate:
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
     entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK]]
@@ -39,7 +39,7 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     entities: list[entity_t.Entity]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entity.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
 from ... import entity_t
@@ -29,7 +29,7 @@ ENDPOINT_PATH = "api/external/entity/external_create_entity"
 @serial_class(
     unconverted_values={"value"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class EntityFieldInitialValue:
     field_ref_name: str
     value: base_t.JsonValue
@@ -37,7 +37,7 @@ class EntityFieldInitialValue:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
     entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK]]
@@ -45,7 +45,7 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     entity: entity_t.Entity
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/get_entities_data.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
 from ... import entity_t
 from ... import field_values_t
@@ -25,21 +25,21 @@ ENDPOINT_PATH = "api/external/entity/external_get_entities_data"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_ids: list[base_t.ObjectId]
     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 EntityDetails:
     entity: entity_t.Entity
     field_values: list[field_values_t.FieldRefIdNameValue]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     entity_details: list[EntityDetails]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/list_entities.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 pkgs.serialization import OpaqueKey
 from ... import base_t
@@ -30,7 +30,7 @@ ENDPOINT_PATH = "api/external/entity/external_list_entities"
 @serial_class(
     unconverted_values={"attributes"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     config_reference: str
     entity_type: typing.Optional[entity_t.EntityType] = None
@@ -43,21 +43,21 @@ class Arguments:
 @serial_class(
     unconverted_values={"column_values"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class EntityResult:
     entity: entity_t.Entity
     column_values: list[base_t.JsonValue]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class ColumnAccess:
     name: str
     table_label: typing.Optional[str]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     columns: list[ColumnAccess]
     results: list[EntityResult]

uncountable/types/api/entity/lock_entity.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 async_batch_t
 from ... import entity_t
 from ... import identifier_t
@@ -24,7 +24,7 @@ ENDPOINT_PATH = "api/external/entity/external_lock_entity"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_key: identifier_t.IdentifierKey
     entity_type: entity_t.EntityType
@@ -32,7 +32,7 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/resolve_entity_ids.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
 from ... import entity_t
 
@@ -24,21 +24,21 @@ ENDPOINT_PATH = "api/external/entity/external_resolve_entity_ids"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_ids: list[typing.Union[str, base_t.ObjectId]]
     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 EntityNames:
     id: typing.Union[str, int]
     name: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     items: list[EntityNames]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/set_values.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 field_values_t
 from ... import response_t
@@ -24,14 +24,14 @@ ENDPOINT_PATH = "api/external/entity/external_set_values"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity: entity_t.Entity
     values: list[field_values_t.ArgumentValueRefName]
 
 
 # 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/entity/transition_entity_phase.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 entity_t
 from ... import identifier_t
@@ -31,7 +31,7 @@ ENDPOINT_PATH = "api/external/entity/transition_entity_phase"
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class TransitionIdentifierPhases:
     type: typing.Literal["phases"] = "phases"
     phase_from_key: identifier_t.IdentifierKey
@@ -42,7 +42,7 @@ class TransitionIdentifierPhases:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class TransitionIdentifierTransition:
     type: typing.Literal["transition"] = "transition"
     transition_key: identifier_t.IdentifierKey
@@ -53,14 +53,14 @@ TransitionIdentifier = typing.Union[TransitionIdentifierPhases, TransitionIdenti
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity: entity_t.Entity
     transition: TransitionIdentifier
 
 
 # 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/entity/unlock_entity.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 async_batch_t
 from ... import entity_t
 from ... import identifier_t
@@ -24,14 +24,14 @@ ENDPOINT_PATH = "api/external/entity/external_unlock_entity"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_key: identifier_t.IdentifierKey
     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 Data(async_batch_t.AsyncBatchActionReturn):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/equipment/associate_equipment_input.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 async_batch_t
 from ... import base_t
 from ... import identifier_t
@@ -24,14 +24,14 @@ ENDPOINT_PATH = "api/external/equipment/associate_equipment_input"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     equipment_key: identifier_t.IdentifierKey
     material_family_ids: list[base_t.ObjectId]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/field_options/upsert_field_options.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] = [
@@ -23,7 +23,7 @@ ENDPOINT_PATH = "api/external/field_options/upsert_field_options"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class FieldOption:
     option_id: str
     option_value: typing.Optional[str] = None
@@ -32,14 +32,14 @@ class FieldOption:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     option_set_key: identifier_t.IdentifierKey
     field_options: list[FieldOption]
 
 
 # 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/id_source/list_id_source.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
 from ... import id_source_t
 
@@ -24,7 +24,7 @@ ENDPOINT_PATH = "api/external/id_source/list_id_source"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     spec: id_source_t.IdSourceSpec
     search_label: str
@@ -33,14 +33,14 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class IdName:
     id: typing.Union[base_t.ObjectId, str]
     name: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     results: list[IdName]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/id_source/match_id_source.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
 from ... import id_source_t
 
@@ -24,21 +24,21 @@ ENDPOINT_PATH = "api/external/id_source/match_id_source"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     spec: id_source_t.IdSourceSpec
     names: list[str]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Match:
     name: str
     ids: list[typing.Union[base_t.ObjectId, str]]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     results: list[Match]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/input_groups/get_input_group_names.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] = [
@@ -23,20 +23,20 @@ ENDPOINT_PATH = "api/external/input_groups/external_get_input_group_names"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Arguments:
     material_family_id: base_t.ObjectId
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleInputGroup:
     input_group_id: base_t.ObjectId
     name: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Data:
     input_groups: list[SimpleInputGroup]
 # DO NOT MODIFY -- This file is generated by type_spec