cognite-toolkit 0.7.45__py3-none-any.whl → 0.7.47__py3-none-any.whl

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_data_post_processing.py

@@ -0,0 +1,49 @@
+from typing import ClassVar
+
+from pydantic import JsonValue
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestUpdateable,
+    ResponseResource,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+
+
+class RobotDataPostProcessing(BaseModelObject):
+    """DataPostprocessing define types of data processing on data captured by the robot.
+    DataPostprocessing enables you to automatically process data captured by the robot.
+
+    Args:
+        name: DataPostProcessing name.
+        external_id: DataPostProcessing external id. Must be unique for the resource type.
+        method: DataPostProcessing method. The method is used to call the right functionality on the robot.
+        input_schema: Schema that defines what inputs are needed for the data postprocessing.
+            The input are values that configure the data postprocessing, e.g max and min values for a gauge.
+        description: Description of DataPostProcessing. Textual description of the DataPostProcessing.
+    """
+
+    external_id: str
+    name: str
+    method: str
+    description: str | None = None
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class RobotDataPostProcessingRequest(RobotDataPostProcessing, RequestUpdateable):
+    """Request resource for creating or updating a DataPostProcessing."""
+
+    non_nullable_fields: ClassVar[frozenset[str]] = frozenset({"input_schema"})
+    input_schema: dict[str, JsonValue] | None = None
+
+
+class RobotDataPostProcessingResponse(RobotDataPostProcessing, ResponseResource[RobotDataPostProcessingRequest]):
+    """Response resource for a DataPostProcessing."""
+
+    # The response always has input_schema
+    input_schema: dict[str, JsonValue]
+
+    def as_request_resource(self) -> RobotDataPostProcessingRequest:
+        return RobotDataPostProcessingRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_frame.py

@@ -0,0 +1,46 @@
+from typing import ClassVar
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestUpdateable,
+    ResponseResource,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+
+from ._common import Transform
+
+
+class RobotFrame(BaseModelObject):
+    """The frames resource represents coordinate frames, which are used to describe how maps are aligned with
+    respect to each other. For example, frames are used to describe the relative position of a context map
+    (e.g., a 3D model of a location) and a robot's navigation map. Frames are aligned with each other through
+    transforms, which consist of a translation (in meters) and rotation (quaternion).
+
+    Args:
+        name: Frame name.
+        external_id: Frame external id. Must be unique for the resource type.
+        transform: Transform of the parent frame to the current frame.
+    """
+
+    external_id: str
+    name: str
+    transform: Transform | None = None
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class RobotFrameRequest(RobotFrame, RequestUpdateable):
+    """Request resource for creating or updating a Frame."""
+
+    non_nullable_fields: ClassVar[frozenset[str]] = frozenset({"transform"})
+
+
+class RobotFrameResponse(RobotFrame, ResponseResource[RobotFrameRequest]):
+    """Response resource for a Frame."""
+
+    created_time: int
+    updated_time: int
+
+    def as_request_resource(self) -> RobotFrameRequest:
+        return RobotFrameRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_location.py

@@ -0,0 +1,36 @@
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+
+
+class RobotLocation(BaseModelObject):
+    """The Locations resource is used to specify the physical location of a robot. Robot missions are defined
+    for a specific location. In addition, the location is used to group Missions and Map resources.
+
+    Args:
+        name: Location name.
+        external_id: Location external id. Must be unique for the resource type.
+        description: Description of Location. Textual description of the Location.
+    """
+
+    external_id: str
+    name: str
+    description: str | None = None
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class RobotLocationRequest(RobotLocation, RequestUpdateable):
+    """Request resource for creating or updating a Location."""
+
+    ...
+
+
+class RobotLocationResponse(RobotLocation, ResponseResource[RobotLocationRequest]):
+    """Response resource for a Location."""
+
+    created_time: int
+    updated_time: int
+
+    def as_request_resource(self) -> RobotLocationRequest:
+        return RobotLocationRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_map.py

@@ -0,0 +1,60 @@
+from typing import ClassVar
+
+from pydantic import JsonValue
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestUpdateable,
+    ResponseResource,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+
+from ._common import MapType
+
+
+class RobotMap(BaseModelObject):
+    """The map resource allows defining both context maps and robot maps of a specific location. A context map is a
+    visual representation of a location, for example, a 3D model, a 2D floor plan, or a point cloud model.
+    A robot map is a representation of where a robot is able to navigate. Maps need to be aligned with respect
+    to each other using coordinate frames.
+
+    Args:
+        name: Map name.
+        external_id: Map external id. Must be unique for the resource type.
+        map_type: Map type.
+        description: Description of Map. Textual description of the Map.
+        frame_external_id: External id of the map's reference frame.
+        data: Map-specific data.
+        location_external_id: External id of the location.
+        scale: Uniform scaling factor, for example, for map unit conversion (centimeter to meter).
+    """
+
+    external_id: str
+    name: str
+    map_type: MapType
+    description: str | None = None
+    frame_external_id: str | None = None
+    data: dict[str, JsonValue] | None = None
+    location_external_id: str | None = None
+    scale: float | None = None
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class RobotMapRequest(RobotMap, RequestUpdateable):
+    """Request resource for creating or updating a RobotMap."""
+
+    non_nullable_fields: ClassVar[frozenset[str]] = frozenset(
+        {"description", "data", "frame_external_id", "location_external_id", "scale"}
+    )
+
+
+class RobotMapResponse(RobotMap, ResponseResource[RobotMapRequest]):
+    """Response resource for a RobotMap."""
+
+    created_time: int
+    updated_time: int
+
+    def as_request_resource(self) -> RobotMapRequest:
+        return RobotMapRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_robot.py

@@ -0,0 +1,53 @@
+from typing import ClassVar
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestUpdateable,
+    ResponseResource,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import NameId
+
+from ._common import RobotType
+
+
+class Robot(BaseModelObject):
+    """Robot contains information for a single robot, including capabilities, type, video streaming setup, etc.
+    These fields are updated every time a robot registers with Robotics Services and do not require manual changes.
+
+    Args:
+        name: Robot name.
+        capabilities: List of external ids for the capabilities the robot can perform.
+        robot_type: Type of robot.
+        data_set_id: The id of the data set this asset belongs to.
+        description: A brief description of the robot.
+        metadata: Custom, application-specific metadata. String key -> String value.
+        location_external_id: External id of the location.
+    """
+
+    name: str
+    capabilities: list[str]
+    robot_type: RobotType
+    data_set_id: int
+    description: str | None = None
+    metadata: dict[str, str] | None = None
+    location_external_id: str | None = None
+
+    def as_id(self) -> NameId:
+        return NameId(name=self.name)
+
+
+class RobotRequest(Robot, RequestUpdateable):
+    """Request resource for creating or updating a Robot."""
+
+    container_fields: ClassVar[frozenset[str]] = frozenset({"metadata"})
+    non_nullable_fields: ClassVar[frozenset[str]] = frozenset({"location_external_id"})
+
+
+class RobotResponse(Robot, ResponseResource[RobotRequest]):
+    """Response resource for a Robot."""
+
+    created_time: int
+    updated_time: int
+
+    def as_request_resource(self) -> RobotRequest:
+        return RobotRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/search_config_resource.py

@@ -0,0 +1,54 @@
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
+
+from .identifiers import Identifier
+
+
+class SearchConfigViewId(Identifier):
+    """Identifier for a view in search configuration."""
+
+    space: str
+    external_id: str
+
+    def __str__(self) -> str:
+        return f"space='{self.space}', externalId='{self.external_id}'"
+
+
+class SearchConfigViewProperty(BaseModelObject):
+    """Configuration for a property in search configuration."""
+
+    property: str
+    disabled: bool | None = None
+    selected: bool | None = None
+    hidden: bool | None = None
+
+
+class SearchConfigBase(BaseModelObject):
+    """Base class for search configuration with common fields."""
+
+    view: SearchConfigViewId
+    use_as_name: str | None = None
+    use_as_description: str | None = None
+    columns_layout: list[SearchConfigViewProperty] | None = None
+    filter_layout: list[SearchConfigViewProperty] | None = None
+    properties_layout: list[SearchConfigViewProperty] | None = None
+
+    def as_id(self) -> SearchConfigViewId:
+        return self.view
+
+
+class SearchConfigRequest(SearchConfigBase, RequestResource):
+    """Request resource for creating/updating search configuration."""
+
+    # This is required when updating an existing search config
+    id: int | None = None
+
+
+class SearchConfigResponse(SearchConfigBase, ResponseResource[SearchConfigRequest]):
+    """Response resource for search configuration."""
+
+    id: int
+    created_time: int
+    last_updated_time: int
+
+    def as_request_resource(self) -> SearchConfigRequest:
+        return SearchConfigRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/securitycategory.py

@@ -1,4 +1,4 @@
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestResource,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/sequence.py

@@ -1,6 +1,6 @@
 from typing import ClassVar, Literal
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/resource_classes/sequence_rows.py

@@ -0,0 +1,56 @@
+from typing import Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
+
+from .identifiers import ExternalId
+
+
+class SequenceColumn(BaseModelObject):
+    """Represents a column in a sequence."""
+
+    external_id: str
+    name: str | None = None
+    description: str | None = None
+    value_type: Literal["STRING", "DOUBLE", "LONG"] | None = None
+    metadata: dict[str, str] | None = None
+
+
+class SequenceRow(BaseModelObject):
+    """Represents a row in a sequence."""
+
+    row_number: int
+    values: list[str | int | float]
+
+
+class SequenceRows(BaseModelObject):
+    """Base class for sequence rows with common fields."""
+
+    rows: list[SequenceRow]
+
+
+class SequenceRowsRequest(SequenceRows, RequestResource):
+    """Request resource for inserting sequence rows."""
+
+    external_id: str
+    columns: list[str]
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class SequenceRowsResponse(SequenceRows, ResponseResource[SequenceRowsRequest]):
+    """Response resource for sequence rows.
+
+    Note: The CDF API for sequence rows typically doesn't return the standard
+    created_time/last_updated_time fields in the same way as other resources.
+    The response contains the sequence identification and the row data.
+    """
+
+    external_id: str | None = None
+    id: int
+    columns: list[SequenceColumn]
+
+    def as_request_resource(self) -> SequenceRowsRequest:
+        dumped = self.dump()
+        dumped["columns"] = [col.external_id for col in self.columns]
+        return SequenceRowsRequest.model_validate(dumped, extra="ignore")

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/simulator_model.py

@@ -2,7 +2,7 @@ from typing import Any, Literal
 
 from pydantic import Field
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import RequestUpdateable, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import RequestUpdateable, ResponseResource
 
 from .identifiers import ExternalId
 

cognite_toolkit/_cdf_tk/client/resource_classes/streamlit_.py

@@ -0,0 +1,71 @@
+from typing import Any, Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestUpdateable,
+    ResponseResource,
+)
+
+from .filemetadata import FileMetadataRequest
+from .identifiers import ExternalId
+
+
+class StreamlitFile(BaseModelObject):
+    """Base class for Streamlit app with common fields."""
+
+    external_id: str
+    name: str
+    creator: str
+    entrypoint: str | None = None
+    description: str | None = None
+    published: bool = False
+    theme: Literal["Light", "Dark"] = "Light"
+    thumbnail: str | None = None
+    data_set_id: int | None = None
+    cognite_toolkit_app_hash: str | None = None
+
+
+class StreamlitRequest(StreamlitFile, RequestUpdateable):
+    """Request resource for creating/updating Streamlit apps."""
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+    def as_file(self) -> FileMetadataRequest:
+        metadata: dict[str, str] = {
+            "creator": self.creator,
+            "name": self.name,
+            "published": str(self.published),
+            "theme": str(self.theme),
+        }
+        if self.description:
+            metadata["description"] = self.description
+        if self.thumbnail:
+            metadata["thumbnail"] = self.thumbnail
+        if self.cognite_toolkit_app_hash:
+            metadata["cdf-toolkit-app-hash"] = self.cognite_toolkit_app_hash
+        if self.entrypoint:
+            metadata["entrypoint"] = self.entrypoint
+        return FileMetadataRequest(
+            external_id=self.external_id,
+            name=f"{self.name}-source.json",
+            data_set_id=self.data_set_id,
+            directory="/streamlit-apps/",
+            metadata=metadata,
+        )
+
+    def dump(self, camel_case: bool = True, exclude_extra: bool = False) -> dict[str, Any]:
+        return self.as_file().dump(camel_case=camel_case, exclude_extra=exclude_extra)
+
+    def as_update(self, mode: Literal["patch", "replace"]) -> dict[str, Any]:
+        return self.as_file().as_update(mode)
+
+
+class StreamlitResponse(StreamlitFile, ResponseResource[StreamlitRequest]):
+    """Response resource for Streamlit apps."""
+
+    created_time: int
+    last_updated_time: int
+
+    def as_request_resource(self) -> StreamlitRequest:
+        return StreamlitRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/streams.py

@@ -10,10 +10,13 @@ from .base import BaseModelObject, BaseResourceList, RequestResource, ResponseRe
 from .identifiers import ExternalId
 
 
-class StreamRequest(RequestResource):
+class Stream(BaseModelObject):
+    external_id: str
+
+
+class StreamRequest(Stream, RequestResource):
     """Stream request resource class."""
 
-    external_id: str
     settings: dict[Literal["template"], dict[Literal["name"], StreamTemplateName]]
 
     def as_id(self) -> ExternalId:
@@ -29,7 +32,6 @@ class StreamRequestList(BaseResourceList[StreamRequest], ResourceRequestListProt
 class LifecycleObject(BaseModelObject):
     """Lifecycle object."""
 
-    hot_phase_duration: str | None = None
     data_deleted_after: str | None = None
     retained_after_soft_delete: str
 
@@ -56,10 +58,9 @@ class StreamSettings(BaseModelObject):
     limits: LimitsObject
 
 
-class StreamResponse(ResponseResource["StreamRequest"]):
+class StreamResponse(Stream, ResponseResource[StreamRequest]):
     """Stream response resource class."""
 
-    external_id: str
     created_time: int
     created_from_template: StreamTemplateName
     type: Literal["Mutable", "Immutable"]

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/timeseries.py

@@ -1,6 +1,6 @@
 from typing import Any, ClassVar, Literal
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/transformation.py

@@ -2,7 +2,7 @@ from typing import Annotated, ClassVar, Literal
 
 from pydantic import Field, JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/workflow.py

@@ -1,4 +1,4 @@
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestResource,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/workflow_trigger.py

@@ -2,7 +2,7 @@ from typing import Annotated, Literal
 
 from pydantic import Field, JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestResource,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/workflow_version.py

@@ -3,7 +3,7 @@ from typing import Annotated, Any, Literal
 from pydantic import Field, JsonValue, field_validator
 from pydantic_core.core_schema import ValidationInfo
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestResource,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/testing.py

@@ -13,8 +13,6 @@ from cognite.client.testing import CogniteClientMock
 from rich.console import Console
 
 from cognite_toolkit._cdf_tk.client._toolkit_client import ToolkitClient
-from cognite_toolkit._cdf_tk.client.api.assets import AssetsAPI
-from cognite_toolkit._cdf_tk.client.api.filemetadata import FileMetadataAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.canvas import CanvasAPI, IndustrialCanvasAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.charts import ChartsAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.dml import DMLAPI
@@ -35,8 +33,13 @@ from cognite_toolkit._cdf_tk.client.api.legacy.robotics import LocationsAPI as R
 from cognite_toolkit._cdf_tk.client.api.legacy.search_config import SearchConfigurationsAPI
 
 from ._toolkit_client import ToolAPI
+from .api.assets import AssetsAPI
+from .api.datasets import DataSetsAPI
 from .api.events import EventsAPI
+from .api.extraction_pipelines import ExtractionPipelinesAPI
+from .api.filemetadata import FileMetadataAPI
 from .api.infield import InfieldAPI, InFieldCDMConfigAPI, InfieldConfigAPI
+from .api.labels import LabelsAPI
 from .api.lookup import (
     AssetLookUpAPI,
     DataSetLookUpAPI,
@@ -59,11 +62,17 @@ from .api.migration import (
 )
 from .api.project import ProjectAPI
 from .api.search import SearchAPI
+from .api.security_categories import SecurityCategoriesAPI
+from .api.sequences import SequencesAPI
 from .api.streams import StreamsAPI
 from .api.three_d import ThreeDAPI, ThreeDModelAPI
 from .api.timeseries import TimeSeriesAPI
 from .api.token import TokenAPI
+from .api.transformations import TransformationsAPI
 from .api.verify import VerifyAPI
+from .api.workflow_triggers import WorkflowTriggersAPI
+from .api.workflow_versions import WorkflowVersionsAPI
+from .api.workflows import WorkflowsAPI
 
 
 class ToolkitClientMock(CogniteClientMock):
@@ -149,6 +158,15 @@ class ToolkitClientMock(CogniteClientMock):
         self.tool.events = MagicMock(spec_set=EventsAPI)
         self.tool.simulators = MagicMock(spec=SimulatorsAPI)
         self.tool.simulators.models = MagicMock(spec_set=SimulatorModelsAPI)
+        self.tool.datasets = MagicMock(spec_set=DataSetsAPI)
+        self.tool.extraction_pipelines = MagicMock(spec_set=ExtractionPipelinesAPI)
+        self.tool.labels = MagicMock(spec_set=LabelsAPI)
+        self.tool.security_categories = MagicMock(spec_set=SecurityCategoriesAPI)
+        self.tool.sequences = MagicMock(spec_set=SequencesAPI)
+        self.tool.transformations = MagicMock(spec_set=TransformationsAPI)
+        self.tool.workflows = MagicMock(spec=WorkflowsAPI)
+        self.tool.workflows.triggers = MagicMock(spec_set=WorkflowTriggersAPI)
+        self.tool.workflows.versions = MagicMock(spec_set=WorkflowVersionsAPI)
 
         self.streams = MagicMock(spec=StreamsAPI)
 

cognite_toolkit/_cdf_tk/commands/_migrate/conversion.py

@@ -16,14 +16,14 @@ from cognite.client.data_classes.data_modeling.views import ViewProperty
 from cognite.client.utils._identifier import InstanceId
 
 from cognite_toolkit._cdf_tk.client import ToolkitClient
-from cognite_toolkit._cdf_tk.client.data_classes.asset import AssetResponse
-from cognite_toolkit._cdf_tk.client.data_classes.event import EventResponse
-from cognite_toolkit._cdf_tk.client.data_classes.filemetadata import FileMetadataResponse
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.migration import (
+from cognite_toolkit._cdf_tk.client.resource_classes.asset import AssetResponse
+from cognite_toolkit._cdf_tk.client.resource_classes.event import EventResponse
+from cognite_toolkit._cdf_tk.client.resource_classes.filemetadata import FileMetadataResponse
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.migration import (
     AssetCentricId,
     ResourceViewMappingApply,
 )
-from cognite_toolkit._cdf_tk.client.data_classes.timeseries import TimeSeriesResponse
+from cognite_toolkit._cdf_tk.client.resource_classes.timeseries import TimeSeriesResponse
 from cognite_toolkit._cdf_tk.utils.collection import flatten_dict_json_path
 from cognite_toolkit._cdf_tk.utils.dtype_conversion import (
     asset_centric_convert_to_primary_property,

cognite_toolkit/_cdf_tk/commands/_migrate/creators.py

@@ -20,7 +20,7 @@ from cognite.client.data_classes.documents import SourceFileProperty
 from cognite.client.data_classes.events import EventProperty
 
 from cognite_toolkit._cdf_tk.client import ToolkitClient
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.apm_config_v1 import APMConfig, APMConfigList
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.apm_config_v1 import APMConfig, APMConfigList
 from cognite_toolkit._cdf_tk.cruds import NodeCRUD, ResourceCRUD, SpaceCRUD
 from cognite_toolkit._cdf_tk.exceptions import ToolkitMissingResourceError, ToolkitRequiredValueError
 from cognite_toolkit._cdf_tk.utils import humanize_collection

cognite_toolkit/_cdf_tk/commands/_migrate/data_classes.py

@@ -11,12 +11,12 @@ from cognite.client.utils._identifier import InstanceId
 from cognite.client.utils._text import to_camel_case
 from pydantic import BaseModel, BeforeValidator, Field, field_validator, model_validator
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestResource
-from cognite_toolkit._cdf_tk.client.data_classes.identifiers import InternalId
-from cognite_toolkit._cdf_tk.client.data_classes.instance_api import InstanceIdentifier
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.instances import InstanceApplyList
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.migration import AssetCentricId
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.pending_instances_ids import PendingInstanceId
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import InternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.instance_api import InstanceIdentifier
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.instances import InstanceApplyList
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.migration import AssetCentricId
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.pending_instances_ids import PendingInstanceId
 from cognite_toolkit._cdf_tk.commands._migrate.default_mappings import (
     ASSET_ANNOTATIONS_ID,
     FILE_ANNOTATIONS_ID,