UncountablePythonSDK 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

uncountable/types/api/recipes/get_recipes_data.py

@@ -10,6 +10,13 @@ from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
 from ... import base as base_t
+from ... import experiment_groups as experiment_groups_t
+from ... import inputs as inputs_t
+from ... import outputs as outputs_t
+from ... import recipe_metadata as recipe_metadata_t
+from ... import recipe_tags as recipe_tags_t
+from ... import users as users_t
+from ... import workflows as workflows_t
 
 __all__: list[str] = [
     "Arguments",
@@ -19,22 +26,12 @@ __all__: list[str] = [
     "ENDPOINT_PATH",
     "Recipe",
     "RecipeInput",
-    "RecipeMetadata",
     "RecipeOutput",
     "RecipeStep",
     "RecipeStepGroup",
     "RecipeStepRelationship",
     "RecipeWorkflowStep",
-    "SimpleExperimentGroup",
-    "SimpleInput",
-    "SimpleOutput",
     "SimpleOutputCondition",
-    "SimpleRecipeMetadataField",
-    "SimpleRecipeTag",
-    "SimpleUnit",
-    "SimpleUser",
-    "SimpleWorkflow",
-    "SimpleWorkflowStep",
 ]
 
 ENDPOINT_METHOD = "GET"
@@ -44,51 +41,10 @@ ENDPOINT_PATH = "api/external/recipes/external_get_recipes_data"
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class Arguments:
-    recipe_ids: typing.Optional[list[int]]
-    project_id: typing.Optional[int]
-    offset: typing.Optional[int]
-    limit: typing.Optional[int]
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleRecipeMetadataField:
-    metadata_id: int
-    name: str
-    quantity_type: str
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    unconverted_values={"quantity_json"},
-    to_string_values={"quantity_dec"},
-)
-@dataclass(kw_only=True)
-class RecipeMetadata:
-    metadata_id: int
-    quantity_dec: Decimal
-    quantity_json: base_t.JsonValue
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleRecipeTag:
-    recipe_tag_id: int
-    name: str
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleExperimentGroup:
-    experiment_group_id: int
-    name: str
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleUnit:
-    unit_id: int
-    name: str
+    recipe_ids: typing.Optional[typing.Optional[list[base_t.ObjectId]]] = None
+    project_id: typing.Optional[typing.Optional[base_t.ObjectId]] = None
+    offset: typing.Optional[typing.Optional[int]] = None
+    limit: typing.Optional[typing.Optional[int]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -98,22 +54,13 @@ class SimpleUnit:
 )
 @dataclass(kw_only=True)
 class RecipeOutput:
-    id: int
-    output_id: int
+    id: base_t.ObjectId
+    output_id: base_t.ObjectId
     replicate_num: int
     quantity_dec: Decimal
     quantity_json: base_t.JsonValue
-    curve_id: int
-    output_condition_id: int
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleOutput:
-    output_id: int
-    name: str
-    unit: typing.Optional[SimpleUnit]
-    quantity_type: str
+    curve_id: base_t.ObjectId
+    output_condition_id: base_t.ObjectId
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -123,7 +70,7 @@ class SimpleOutput:
 )
 @dataclass(kw_only=True)
 class ConditionParameterValue:
-    condition_parameter_id: int
+    condition_parameter_id: base_t.ObjectId
     condition_parameter_name: str
     quantity_dec: Decimal
     quantity_json: base_t.JsonValue
@@ -132,25 +79,10 @@ class ConditionParameterValue:
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class SimpleOutputCondition:
-    output_condition_id: int
+    output_condition_id: base_t.ObjectId
     condition_parameter_values: list[ConditionParameterValue]
 
 
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleWorkflowStep:
-    workflow_step_id: int
-    name: str
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleWorkflow:
-    workflow_id: int
-    name: str
-    workflow_steps: list[SimpleWorkflowStep]
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     unconverted_values={"actual_quantity_json", "quantity_json"},
@@ -158,9 +90,9 @@ class SimpleWorkflow:
 )
 @dataclass(kw_only=True)
 class RecipeInput:
-    input_id: int
-    input_lot_recipe_id: typing.Optional[int]
-    recipe_step_id: int
+    input_id: base_t.ObjectId
+    input_lot_recipe_id: typing.Optional[base_t.ObjectId]
+    recipe_step_id: base_t.ObjectId
     quantity_dec: Decimal
     quantity_json: base_t.JsonValue
     actual_quantity_dec: Decimal
@@ -169,21 +101,11 @@ class RecipeInput:
     behavior: str
 
 
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleInput:
-    input_id: int
-    quantity_type: str
-    name: str
-    is_parameter: bool
-    intermediate_recipe_id: typing.Optional[int]
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class RecipeStep:
     name: typing.Optional[str]
-    recipe_step_id: int
+    recipe_step_id: base_t.ObjectId
     recipe_step_number: int
     instructions: typing.Optional[str] = None
 
@@ -198,19 +120,11 @@ class RecipeStepGroup:
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class RecipeWorkflowStep:
-    recipe_workflow_step_id: int
-    workflow_step_id: int
+    recipe_workflow_step_id: base_t.ObjectId
+    workflow_step_id: base_t.ObjectId
     recipe_step_groups: list[RecipeStepGroup]
 
 
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class SimpleUser:
-    user_id: int
-    display_name: str
-    email: str
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     to_string_values={"actual_quantity_dec", "quantity_dec"},
@@ -228,20 +142,20 @@ class RecipeStepRelationship:
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class Recipe:
-    recipe_id: int
+    recipe_id: base_t.ObjectId
     create_datetime: str
     last_modified_datetime: str
     name: str
     notes: typing.Optional[str]
-    workflow_id: int
-    metadata: list[RecipeMetadata]
+    workflow_id: base_t.ObjectId
+    metadata: list[recipe_metadata_t.RecipeMetadata]
     inputs: list[RecipeInput]
     outputs: list[RecipeOutput]
     workflow_steps: list[RecipeWorkflowStep]
-    tag_ids: list[int]
-    experiment_group_ids: list[int]
+    tag_ids: list[base_t.ObjectId]
+    experiment_group_ids: list[base_t.ObjectId]
     step_relationships: list[RecipeStepRelationship]
-    creating_user_id: typing.Optional[int] = None
+    creating_user_id: typing.Optional[base_t.ObjectId] = None
     barcode_value: typing.Optional[str] = None
 
 
@@ -249,12 +163,12 @@ class Recipe:
 @dataclass(kw_only=True)
 class Data:
     recipes: list[Recipe]
-    workflows: list[SimpleWorkflow]
-    metadata: list[SimpleRecipeMetadataField]
-    inputs: list[SimpleInput]
-    outputs: list[SimpleOutput]
+    workflows: list[workflows_t.SimpleWorkflow]
+    metadata: list[recipe_metadata_t.SimpleRecipeMetadataField]
+    inputs: list[inputs_t.SimpleInput]
+    outputs: list[outputs_t.SimpleOutput]
     output_conditions: list[SimpleOutputCondition]
-    users: list[SimpleUser]
-    recipe_tags: list[SimpleRecipeTag]
-    experiment_groups: list[SimpleExperimentGroup]
+    users: list[users_t.SimpleUser]
+    recipe_tags: list[recipe_tags_t.SimpleRecipeTag]
+    experiment_groups: list[experiment_groups_t.SimpleExperimentGroup]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_inputs.py

@@ -9,6 +9,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
+from ... import base as base_t
 
 __all__: list[str] = [
     "Arguments",
@@ -28,9 +29,9 @@ ENDPOINT_PATH = "api/external/recipes/external_set_recipe_inputs"
 )
 @dataclass(kw_only=True)
 class RecipeInputValue:
-    recipe_id: int
-    input_id: int
-    recipe_step_id: typing.Optional[typing.Optional[int]] = None
+    recipe_id: base_t.ObjectId
+    input_id: base_t.ObjectId
+    recipe_step_id: typing.Optional[typing.Optional[base_t.ObjectId]] = None
     value_numeric: typing.Optional[Decimal] = None
     value_str: typing.Optional[str] = None
     set_actual_value: typing.Optional[bool] = None

uncountable/types/api/recipes/set_recipe_outputs.py

@@ -9,6 +9,8 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
+from ... import base as base_t
+from ... import response as response_t
 
 __all__: list[str] = [
     "Arguments",
@@ -36,10 +38,10 @@ class CurveValues:
 )
 @dataclass(kw_only=True)
 class RecipeOutputValue:
-    recipe_id: int
-    output_id: int
+    recipe_id: base_t.ObjectId
+    output_id: base_t.ObjectId
     experiment_num: int
-    condition_id: typing.Optional[int] = None
+    condition_id: typing.Optional[base_t.ObjectId] = None
     value_numeric: typing.Optional[Decimal] = None
     value_str: typing.Optional[str] = None
     value_curve: typing.Optional[CurveValues] = None
@@ -52,10 +54,7 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    parse_require={"status"},
-)
 @dataclass(kw_only=True)
-class Data:
-    status: typing.Literal["ok"] = "ok"
+class Data(response_t.Response):
+    pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/calculations.py

@@ -0,0 +1,23 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from dataclasses import dataclass
+from . import base as base_t
+
+__all__: list[str] = [
+    "Calculation",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Calculation:
+    id: base_t.ObjectId
+    name: str
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -64,7 +64,7 @@ class ClientMethods(ABC):
     def create_entities(
         self,
         *,
-        definition_id: int,
+        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.TASK], typing.Literal[entity_t.EntityType.PROJECT]],
         entities_to_create: list[create_entities_t.EntityToCreate],
     ) -> create_entities_t.Data:
@@ -84,7 +84,7 @@ class ClientMethods(ABC):
     def create_entity(
         self,
         *,
-        definition_id: int,
+        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.TASK], typing.Literal[entity_t.EntityType.PROJECT]],
         field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None,
     ) -> create_entity_t.Data:
@@ -120,9 +120,9 @@ class ClientMethods(ABC):
     def create_recipes(
         self,
         *,
-        material_family_id: int,
+        material_family_id: base_t.ObjectId,
         recipe_definitions: list[create_recipes_t.CreateRecipeDefinition],
-        project_id: typing.Optional[int] = None,
+        project_id: typing.Optional[base_t.ObjectId] = None,
     ) -> create_recipes_t.Data:
         args = create_recipes_t.Arguments(
             material_family_id=material_family_id,
@@ -156,7 +156,7 @@ class ClientMethods(ABC):
     def get_curve(
         self,
         *,
-        recipe_output_id: int,
+        recipe_output_id: base_t.ObjectId,
     ) -> get_curve_t.Data:
         args = get_curve_t.Arguments(
             recipe_output_id=recipe_output_id,
@@ -172,8 +172,8 @@ class ClientMethods(ABC):
     def get_entities_data(
         self,
         *,
-        entity_ids: list[int],
-        entity_type: str,
+        entity_ids: list[base_t.ObjectId],
+        entity_type: entity_t.EntityType,
     ) -> get_entities_data_t.Data:
         args = get_entities_data_t.Arguments(
             entity_ids=entity_ids,
@@ -190,11 +190,11 @@ class ClientMethods(ABC):
     def get_input_data(
         self,
         *,
-        material_family_id: int,
-        input_ids: typing.Optional[list[int]],
-        offset: typing.Optional[int],
+        material_family_id: base_t.ObjectId,
+        input_ids: typing.Optional[list[base_t.ObjectId]],
         is_parameter: typing.Optional[bool],
-        limit: typing.Optional[int],
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> get_input_data_t.Data:
         args = get_input_data_t.Arguments(
             material_family_id=material_family_id,
@@ -214,7 +214,7 @@ class ClientMethods(ABC):
     def get_input_group_names(
         self,
         *,
-        material_family_id: int,
+        material_family_id: base_t.ObjectId,
     ) -> get_input_group_names_t.Data:
         args = get_input_group_names_t.Arguments(
             material_family_id=material_family_id,
@@ -230,10 +230,10 @@ class ClientMethods(ABC):
     def get_input_names(
         self,
         *,
-        material_family_id: int,
-        offset: typing.Optional[int],
+        material_family_id: base_t.ObjectId,
         is_parameter: typing.Optional[bool],
-        limit: typing.Optional[int],
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> get_input_names_t.Data:
         args = get_input_names_t.Arguments(
             material_family_id=material_family_id,
@@ -252,7 +252,7 @@ class ClientMethods(ABC):
     def get_inputs_data(
         self,
         *,
-        input_ids: list[int],
+        input_ids: list[base_t.ObjectId],
     ) -> get_inputs_data_t.Data:
         args = get_inputs_data_t.Arguments(
             input_ids=input_ids,
@@ -268,8 +268,8 @@ class ClientMethods(ABC):
     def get_output_data(
         self,
         *,
-        material_family_id: int,
-        output_ids: typing.Optional[list[int]] = None,
+        material_family_id: base_t.ObjectId,
+        output_ids: typing.Optional[list[base_t.ObjectId]] = None,
         offset: typing.Optional[int] = None,
         limit: typing.Optional[int] = None,
     ) -> get_output_data_t.Data:
@@ -290,7 +290,7 @@ class ClientMethods(ABC):
     def get_output_names(
         self,
         *,
-        material_family_id: int,
+        material_family_id: base_t.ObjectId,
         offset: typing.Optional[int] = None,
         limit: typing.Optional[int] = None,
     ) -> get_output_names_t.Data:
@@ -311,7 +311,7 @@ class ClientMethods(ABC):
         self,
         *,
         all_material_families: bool,
-        material_family_id: typing.Optional[int],
+        material_family_id: typing.Optional[base_t.ObjectId],
     ) -> get_projects_t.Data:
         args = get_projects_t.Arguments(
             all_material_families=all_material_families,
@@ -328,9 +328,9 @@ class ClientMethods(ABC):
     def get_projects_data(
         self,
         *,
-        project_ids: list[int],
-        offset: typing.Optional[int],
-        limit: typing.Optional[int],
+        project_ids: list[base_t.ObjectId],
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> get_projects_data_t.Data:
         args = get_projects_data_t.Arguments(
             project_ids=project_ids,
@@ -348,8 +348,8 @@ class ClientMethods(ABC):
     def get_recipe_calculations(
         self,
         *,
-        recipe_ids: list[int],
-        calculation_ids: typing.Optional[list[int]] = None,
+        recipe_ids: list[base_t.ObjectId],
+        calculation_ids: typing.Optional[list[base_t.ObjectId]] = None,
     ) -> get_recipe_calculations_t.Data:
         args = get_recipe_calculations_t.Arguments(
             recipe_ids=recipe_ids,
@@ -366,7 +366,7 @@ class ClientMethods(ABC):
     def get_recipe_links(
         self,
         *,
-        recipe_ids: list[int],
+        recipe_ids: list[base_t.ObjectId],
         depth: int = 1,
         link_types: typing.Optional[list[recipe_links_t.RecipeLinkType]],
     ) -> get_recipe_links_t.Data:
@@ -386,10 +386,10 @@ class ClientMethods(ABC):
     def get_recipe_metadata_data(
         self,
         *,
-        material_family_id: int,
-        recipe_metadata_ids: typing.Optional[list[int]],
-        offset: typing.Optional[int],
-        limit: typing.Optional[int],
+        material_family_id: base_t.ObjectId,
+        recipe_metadata_ids: typing.Optional[list[base_t.ObjectId]],
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> get_recipe_metadata_data_t.Data:
         args = get_recipe_metadata_data_t.Arguments(
             material_family_id=material_family_id,
@@ -408,7 +408,7 @@ class ClientMethods(ABC):
     def get_recipe_names(
         self,
         *,
-        project_id: int,
+        project_id: base_t.ObjectId,
     ) -> get_recipe_names_t.Data:
         args = get_recipe_names_t.Arguments(
             project_id=project_id,
@@ -424,7 +424,7 @@ class ClientMethods(ABC):
     def get_recipe_output_metadata(
         self,
         *,
-        recipe_output_ids: list[int],
+        recipe_output_ids: list[base_t.ObjectId],
     ) -> get_recipe_output_metadata_t.Data:
         args = get_recipe_output_metadata_t.Arguments(
             recipe_output_ids=recipe_output_ids,
@@ -440,10 +440,10 @@ class ClientMethods(ABC):
     def get_recipes_data(
         self,
         *,
-        recipe_ids: typing.Optional[list[int]],
-        project_id: typing.Optional[int],
-        offset: typing.Optional[int],
-        limit: typing.Optional[int],
+        recipe_ids: typing.Optional[typing.Optional[list[base_t.ObjectId]]] = None,
+        project_id: typing.Optional[typing.Optional[base_t.ObjectId]] = None,
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> get_recipes_data_t.Data:
         args = get_recipes_data_t.Arguments(
             recipe_ids=recipe_ids,
@@ -462,11 +462,11 @@ class ClientMethods(ABC):
     def list_entities(
         self,
         *,
-        entity_type: str,
+        entity_type: entity_t.EntityType,
         config_reference: str,
-        offset: typing.Optional[int],
-        limit: typing.Optional[int],
         attributes: typing.Optional[dict[OpaqueKey, base_t.JsonValue]] = None,
+        offset: typing.Optional[typing.Optional[int]] = None,
+        limit: typing.Optional[typing.Optional[int]] = None,
     ) -> list_entities_t.Data:
         args = list_entities_t.Arguments(
             entity_type=entity_type,
@@ -486,8 +486,8 @@ class ClientMethods(ABC):
     def resolve_entity_ids(
         self,
         *,
-        entity_ids: list[typing.Union[str, int]],
-        entity_type: str,
+        entity_ids: list[typing.Union[str, base_t.ObjectId]],
+        entity_type: entity_t.EntityType,
     ) -> resolve_entity_ids_t.Data:
         args = resolve_entity_ids_t.Arguments(
             entity_ids=entity_ids,
@@ -568,8 +568,8 @@ class ClientMethods(ABC):
     def set_values(
         self,
         *,
-        entity: set_values_t.Entity,
-        values: list[set_values_t.EntityFieldValues],
+        entity: entity_t.Entity,
+        values: list[field_values_t.ArgumentValueRefName],
     ) -> set_values_t.Data:
         args = set_values_t.Arguments(
             entity=entity,

uncountable/types/curves.py

@@ -0,0 +1,47 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
+from dataclasses import dataclass
+from pkgs.serialization import serial_class
+from . import base as base_t
+
+__all__: list[str] = [
+    "Curve",
+    "CurveAxisType",
+    "CurveValue",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class CurveAxisType(StrEnum):
+    NUMERIC = "numeric"
+    TIMESTAMP = "timestamp"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Curve:
+    id: base_t.ObjectId
+    x_type: typing.Union[typing.Literal[CurveAxisType.NUMERIC], typing.Literal[CurveAxisType.TIMESTAMP]]
+    x_values: list[CurveValue]
+    y_type: typing.Union[typing.Literal[CurveAxisType.NUMERIC]]
+    y_values: list[CurveValue]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    to_string_values={"quantity"},
+)
+@dataclass(kw_only=True)
+class CurveValue:
+    index: int
+    quantity: typing.Optional[Decimal] = None
+    quantity_timestamp: typing.Optional[datetime.datetime] = None
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/entity.py

@@ -10,6 +10,7 @@ from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
 from pkgs.serialization import serial_string_enum
+from . import base as base_t
 
 __all__: list[str] = [
     "Entity",
@@ -114,6 +115,7 @@ __all__: list[str] = [
         "timesheet_entry": "Timesheet Entry",
         "training_run": "Analyze Experiment",
         "training_set": "Training Set",
+        "time_series": "Time Series",
         "units": "Units",
         "user_favorite": "User Favorite",
         "user": "User",
@@ -236,6 +238,7 @@ class EntityType(StrEnum):
     TIMESHEET_ENTRY = "timesheet_entry"
     TRAINING_RUN = "training_run"
     TRAINING_SET = "training_set"
+    TIME_SERIES = "time_series"
     UNITS = "units"
     USER_FAVORITE = "user_favorite"
     USER = "user"
@@ -266,6 +269,6 @@ class EntityType(StrEnum):
 # DO NOT MODIFY -- This file is generated by type_spec
 @dataclass(kw_only=True)
 class Entity:
-    id: int
+    id: base_t.ObjectId
     type: EntityType
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/experiment_groups.py

@@ -0,0 +1,23 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from dataclasses import dataclass
+from . import base as base_t
+
+__all__: list[str] = [
+    "SimpleExperimentGroup",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class SimpleExperimentGroup:
+    experiment_group_id: base_t.ObjectId
+    name: str
+# DO NOT MODIFY -- This file is generated by type_spec