cognite-neat 0.92.3__py3-none-any.whl → 0.93.0__py3-none-any.whl

cognite/neat/{utils → _utils}/auth.py

@@ -9,7 +9,7 @@ from cognite.client import ClientConfig, CogniteClient
 from cognite.client.credentials import CredentialProvider, OAuthClientCredentials, OAuthInteractive, Token
 
 from cognite.neat import _version
-from cognite.neat.utils.auxiliary import local_import
+from cognite.neat._utils.auxiliary import local_import
 
 __all__ = ["get_cognite_client", "EnvironmentVariables"]
 
@@ -18,7 +18,18 @@ _VALID_LOGIN_FLOWS = get_args(_LOGIN_FLOW)
 _CLIENT_NAME = f"CogniteNeat:{_version.__version__}"
 
 
-def get_cognite_client(env_file_name: str = ".env") -> CogniteClient:
+def get_cognite_client(env_file_name: str) -> CogniteClient:
+    """Instantiate a CogniteClient using environment variables. If the environment variables are not set, the user will
+    be prompted to enter them.
+
+    Args:
+        env_file_name: The name of the .env file to look for in the repository root. If the file is found, the variables
+            will be loaded from the file. If the file is not found, the user will be prompted to enter the variables.
+
+    Returns:
+        CogniteClient: A CogniteClient instance.
+
+    """
     if not env_file_name.endswith(".env"):
         raise ValueError("env_file_name must end with '.env'")
     with suppress(KeyError):
@@ -320,5 +331,5 @@ def _env_in_gitignore(repo_root: Path, env_file_name: str) -> bool:
 
 
 if __name__ == "__main__":
-    c = get_cognite_client()
+    c = get_cognite_client(".env")
     print(c.iam.token.inspect())

cognite/neat/{utils → _utils}/auxiliary.py

@@ -10,7 +10,7 @@ from types import ModuleType
 
 from cognite.client.exceptions import CogniteDuplicatedError, CogniteReadTimeout
 
-from cognite.neat.issues.errors import NeatImportError
+from cognite.neat._issues.errors import NeatImportError
 
 
 def local_import(module: str, extra: str) -> ModuleType:

cognite/neat/{utils → _utils}/cdf/loaders/_data_modeling.py

@@ -31,6 +31,7 @@ from cognite.client.data_classes.data_modeling import (
 from cognite.client.data_classes.data_modeling.ids import (
     ContainerId,
     DataModelId,
+    NodeId,
     ViewId,
 )
 from cognite.client.exceptions import CogniteAPIError
@@ -115,9 +116,14 @@ class SpaceLoader(DataModelingLoader[str, SpaceApply, Space, SpaceApplyList, Spa
         nodes = self.client.data_modeling.instances.list(
             "node", limit=-1, filter=filters.Equals(["node", "space"], space)
         )
-        if nodes:
-            instances = self.client.data_modeling.instances.delete(nodes=nodes.as_ids())
+        node_types = {NodeId(node.type.space, node.type.external_id) for node in nodes if node.type}
+        node_data = set(nodes.as_ids()) - node_types
+        if node_data:
+            instances = self.client.data_modeling.instances.delete(nodes=list(node_data))
             print(f"Deleted {len(instances.nodes)} nodes")
+        if node_types:
+            instances = self.client.data_modeling.instances.delete(nodes=list(node_types))
+            print(f"Deleted {len(instances.nodes)} node types")
         views = self.client.data_modeling.views.list(limit=-1, space=space)
         if views:
             deleted_views = self.client.data_modeling.views.delete(views.as_ids())

cognite/neat/{utils → _utils}/cdf/loaders/_ingestion.py

@@ -15,7 +15,7 @@ from cognite.client.data_classes import (
 from cognite.client.exceptions import CogniteAPIError
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite.neat.utils.cdf.data_classes import RawTable, RawTableID, RawTableList, RawTableWrite, RawTableWriteList
+from cognite.neat._utils.cdf.data_classes import RawTable, RawTableID, RawTableList, RawTableWrite, RawTableWriteList
 
 from ._base import ResourceLoader
 

cognite/neat/{utils → _utils}/upload.py

@@ -3,8 +3,8 @@ from dataclasses import dataclass, field
 from functools import total_ordering
 from typing import Any, Generic
 
+from cognite.neat._issues import NeatIssueList
 from cognite.neat._shared import T_ID, NeatList, NeatObject
-from cognite.neat.issues import NeatIssueList
 
 
 @total_ordering

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.92.3"
+__version__ = "0.93.0"

cognite/neat/_workflows/__init__.py

@@ -0,0 +1,17 @@
+from cognite.neat._workflows.base import BaseWorkflow
+from cognite.neat._workflows.manager import WorkflowManager
+from cognite.neat._workflows.model import (
+    FlowMessage,
+    WorkflowFullStateReport,
+    WorkflowStepDefinition,
+    WorkflowStepEvent,
+)
+
+__all__ = [
+    "BaseWorkflow",
+    "WorkflowStepDefinition",
+    "WorkflowManager",
+    "WorkflowStepEvent",
+    "WorkflowFullStateReport",
+    "FlowMessage",
+]

cognite/neat/{workflows → _workflows}/base.py

@@ -11,13 +11,13 @@ import yaml
 from cognite.client import ClientConfig, CogniteClient
 from prometheus_client import Gauge
 
-from cognite.neat.app.monitoring.metrics import NeatMetricsCollector
-from cognite.neat.config import Config
-from cognite.neat.issues.errors import WorkflowConfigurationNotSetError, WorkflowStepOutputError
-from cognite.neat.utils.auxiliary import retry_decorator
-from cognite.neat.workflows import cdf_store, utils
-from cognite.neat.workflows.cdf_store import CdfStore
-from cognite.neat.workflows.model import (
+from cognite.neat._app.monitoring.metrics import NeatMetricsCollector
+from cognite.neat._config import Config
+from cognite.neat._issues.errors import WorkflowConfigurationNotSetError, WorkflowStepOutputError
+from cognite.neat._utils.auxiliary import retry_decorator
+from cognite.neat._workflows import cdf_store, utils
+from cognite.neat._workflows.cdf_store import CdfStore
+from cognite.neat._workflows.model import (
     FlowMessage,
     StepExecutionStatus,
     StepType,
@@ -31,9 +31,9 @@ from cognite.neat.workflows.model import (
     WorkflowStepEvent,
     WorkflowSystemComponent,
 )
-from cognite.neat.workflows.steps.step_model import DataContract
-from cognite.neat.workflows.steps_registry import StepsRegistry
-from cognite.neat.workflows.tasks import WorkflowTaskBuilder
+from cognite.neat._workflows.steps.step_model import DataContract
+from cognite.neat._workflows.steps_registry import StepsRegistry
+from cognite.neat._workflows.tasks import WorkflowTaskBuilder
 
 summary_metrics = Gauge("neat_workflow_summary_metrics", "Workflow execution summary metrics", ["wf_name", "name"])
 steps_metrics = Gauge("neat_workflow_steps_metrics", "Workflow step level metrics", ["wf_name", "step_name", "name"])

cognite/neat/{workflows → _workflows}/cdf_store.py

@@ -11,9 +11,9 @@ from cognite.client.data_classes import Event, FileMetadataUpdate
 from fastapi.encoders import jsonable_encoder
 from pydantic import BaseModel
 
-from cognite.neat.issues.errors import WorkflowConfigurationNotSetError
-from cognite.neat.workflows.model import WorkflowFullStateReport, WorkflowState, WorkflowStepEvent
-from cognite.neat.workflows.utils import get_file_hash
+from cognite.neat._issues.errors import WorkflowConfigurationNotSetError
+from cognite.neat._workflows.model import WorkflowFullStateReport, WorkflowState, WorkflowStepEvent
+from cognite.neat._workflows.utils import get_file_hash
 
 
 class NeatCdfResource(BaseModel):

cognite/neat/{workflows → _workflows}/examples/Export_DMS/workflow.yaml

@@ -1,89 +1,89 @@
-configs: []
-description: null
-implementation_module: null
-name: Export DMS
-steps:
--   complex_configs: {}
-    configs: {}
-    description: null
-    enabled: true
-    id: step_861205
-    label: Upload Rules Spreadsheet
-    max_retries: 0
-    method: null
-    params:
-        file_type: rules
-    retry_delay: 3
-    stype: file_uploader
-    system_component_id: null
-    transition_to:
-    - step_295479
-    trigger: true
-    ui_config:
-        pos_x: 629
-        pos_y: 57
--   complex_configs: {}
-    configs:
-        File name: ''
-        Report formatter: BasicHTML
-        Role: infer
-    description: null
-    enabled: true
-    id: step_295479
-    label: Validate
-    max_retries: 0
-    method: ExcelToRules
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to:
-    - step_50885
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 161
--   complex_configs:
-        Components:
-            containers: true
-            data_models: true
-            spaces: true
-            views: true
-    configs:
-        Dry run: 'False'
-        Existing component handling: update
-        Multi-space components create: 'True'
-    description: null
-    enabled: true
-    id: step_50885
-    label: Export Data Model to CDF
-    max_retries: 0
-    method: RulesToDMS
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to:
-    - step_171560
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 243
--   complex_configs: {}
-    configs: {}
-    description: null
-    enabled: true
-    id: step_171560
-    label: Export Transformations
-    max_retries: 0
-    method: RulesToCDFTransformations
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to: []
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 342
-system_components: []
+configs: []
+description: null
+implementation_module: null
+name: Export DMS
+steps:
+-   complex_configs: {}
+    configs: {}
+    description: null
+    enabled: true
+    id: step_861205
+    label: Upload Rules Spreadsheet
+    max_retries: 0
+    method: null
+    params:
+        file_type: rules
+    retry_delay: 3
+    stype: file_uploader
+    system_component_id: null
+    transition_to:
+    - step_295479
+    trigger: true
+    ui_config:
+        pos_x: 629
+        pos_y: 57
+-   complex_configs: {}
+    configs:
+        File name: ''
+        Report formatter: BasicHTML
+        Role: infer
+    description: null
+    enabled: true
+    id: step_295479
+    label: Validate
+    max_retries: 0
+    method: ExcelToRules
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to:
+    - step_50885
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 161
+-   complex_configs:
+        Components:
+            containers: true
+            data_models: true
+            spaces: true
+            views: true
+    configs:
+        Dry run: 'False'
+        Existing component handling: update
+        Multi-space components create: 'True'
+    description: null
+    enabled: true
+    id: step_50885
+    label: Export Data Model to CDF
+    max_retries: 0
+    method: RulesToDMS
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to:
+    - step_171560
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 243
+-   complex_configs: {}
+    configs: {}
+    description: null
+    enabled: true
+    id: step_171560
+    label: Export Transformations
+    max_retries: 0
+    method: RulesToCDFTransformations
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to: []
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 342
+system_components: []

cognite/neat/{workflows → _workflows}/manager.py

@@ -9,12 +9,12 @@ from cognite.client import CogniteClient
 from prometheus_client import Gauge
 from pydantic import BaseModel
 
-from cognite.neat.config import Config
-from cognite.neat.workflows import BaseWorkflow
-from cognite.neat.workflows.base import WorkflowDefinition
-from cognite.neat.workflows.model import FlowMessage, InstanceStartMethod, WorkflowState, WorkflowStepDefinition
-from cognite.neat.workflows.steps_registry import StepsRegistry
-from cognite.neat.workflows.tasks import WorkflowTaskBuilder
+from cognite.neat._config import Config
+from cognite.neat._workflows import BaseWorkflow
+from cognite.neat._workflows.base import WorkflowDefinition
+from cognite.neat._workflows.model import FlowMessage, InstanceStartMethod, WorkflowState, WorkflowStepDefinition
+from cognite.neat._workflows.steps_registry import StepsRegistry
+from cognite.neat._workflows.tasks import WorkflowTaskBuilder
 
 live_workflow_instances = Gauge("neat_workflow_live_instances", "Count of live workflow instances", ["itype"])
 

cognite/neat/{workflows → _workflows}/steps/data_contracts.py

@@ -9,14 +9,14 @@ from cognite.client.data_classes import (
 )
 from cognite.client.data_classes.data_modeling import EdgeApply, NodeApply
 
-from cognite.neat.rules.models import (
+from cognite.neat._rules.models import (
     AssetRules,
     DMSRules,
     DomainRules,
     InformationRules,
 )
-from cognite.neat.store import NeatGraphStore
-from cognite.neat.workflows.steps.step_model import DataContract
+from cognite.neat._store import NeatGraphStore
+from cognite.neat._workflows.steps.step_model import DataContract
 
 
 class MultiRuleData(DataContract):

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_extractor.py

@@ -2,16 +2,13 @@ import json
 from pathlib import Path
 from typing import ClassVar, cast
 
-from rdflib import URIRef
-
-from cognite.neat.constants import DEFAULT_NAMESPACE
-from cognite.neat.graph.extractors import RdfFileExtractor
-from cognite.neat.graph.extractors._mock_graph_generator import MockGraphGenerator
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.rules._shared import DMSRules, InformationRules
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import MultiRuleData, NeatGraph
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._graph.extractors import RdfFileExtractor
+from cognite.neat._graph.extractors._mock_graph_generator import MockGraphGenerator
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._rules._shared import DMSRules, InformationRules
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import MultiRuleData, NeatGraph
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = ["GraphFromRdfFile", "GraphFromMockData"]
 
@@ -83,25 +80,7 @@ class GraphFromRdfFile(Step):
             name="File path",
             value="source-graphs/source-graph-dump.xml",
             label="File name of source graph data dump in RDF format",
-        ),
-        Configurable(
-            name="MIME type",
-            value="application/rdf+xml",
-            label="MIME type of file containing RDF graph",
-            options=[
-                "application/rdf+xml",
-                "text/turtle",
-                "application/n-triples",
-                "application/n-quads",
-                "application/trig",
-            ],
-        ),
-        Configurable(
-            name="Add base URI",
-            value="True",
-            label="Whether to add base URI to graph in case if entity ids are relative",
-            options=["True", "False"],
-        ),
+        )
     ]
 
     def run(self, store: NeatGraph) -> FlowMessage:  # type: ignore[override, syntax]
@@ -112,8 +91,6 @@ class GraphFromRdfFile(Step):
             store.graph.write(
                 RdfFileExtractor(  # type: ignore[abstract]
                     filepath=self.data_store_path / Path(source_file),
-                    mime_type=self.configs["MIME type"],  # type: ignore[arg-type]
-                    base_uri=(URIRef(DEFAULT_NAMESPACE) if self.configs["Add base URI"] == "True" else None),
                 )
             )
 

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_loader.py

@@ -2,10 +2,10 @@ import time
 from pathlib import Path
 from typing import ClassVar
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import NeatGraph
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.steps.data_contracts import NeatGraph
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = [
     "GraphToRdfFile",

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_store.py

@@ -1,12 +1,12 @@
 from typing import ClassVar
 
-from cognite.neat.store import NeatGraphStore
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import (
+from cognite.neat._store import NeatGraphStore
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.steps.data_contracts import (
     MultiRuleData,
     NeatGraph,
 )
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = ["GraphStoreConfiguration"]
 

cognite/neat/{workflows → _workflows}/steps/lib/current/rules_exporter.py

@@ -2,20 +2,20 @@ import time
 from pathlib import Path
 from typing import ClassVar, Literal, cast
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.rules import exporters
-from cognite.neat.rules._shared import DMSRules, InformationRules, VerifiedRules
-from cognite.neat.rules.models import AssetRules, RoleTypes
-from cognite.neat.rules.transformers import (
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._rules import exporters
+from cognite.neat._rules._shared import DMSRules, InformationRules, VerifiedRules
+from cognite.neat._rules.models import AssetRules, RoleTypes
+from cognite.neat._rules.transformers import (
     AssetToInformation,
     DMSToInformation,
     InformationToAsset,
     InformationToDMS,
     RulesPipeline,
 )
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import CogniteClient, MultiRuleData
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import CogniteClient, MultiRuleData
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = [
     "RulesToDMS",

cognite/neat/{workflows → _workflows}/steps/lib/current/rules_importer.py

@@ -5,16 +5,16 @@ from typing import ClassVar
 from cognite.client import CogniteClient
 from cognite.client.data_classes.data_modeling import DataModelId
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.issues.formatters import FORMATTER_BY_NAME
-from cognite.neat.rules import importers
-from cognite.neat.rules._shared import InputRules
-from cognite.neat.rules.models import RoleTypes
-from cognite.neat.rules.models.entities import DataModelEntity, DMSUnknownEntity
-from cognite.neat.rules.transformers import ImporterPipeline
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import MultiRuleData
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._issues.formatters import FORMATTER_BY_NAME
+from cognite.neat._rules import importers
+from cognite.neat._rules._shared import InputRules
+from cognite.neat._rules.models import RoleTypes
+from cognite.neat._rules.models.entities import DataModelEntity, DMSUnknownEntity
+from cognite.neat._rules.transformers import ImporterPipeline
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import MultiRuleData
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 CATEGORY = __name__.split(".")[-1].replace("_", " ").title()
 
@@ -145,7 +145,7 @@ class OntologyToRules(Step):
         if role != "infer" and role is not None:
             role_enum = RoleTypes[role]
 
-        ontology_importer = importers.OWLImporter(filepath=rules_file_path)
+        ontology_importer = importers.OWLImporter.from_file(filepath=rules_file_path)
         result = ImporterPipeline.try_verify(ontology_importer, role_enum)
 
         if result.rules is None:
@@ -216,7 +216,7 @@ class IMFToRules(Step):
         if role != "infer" and role is not None:
             role_enum = RoleTypes[role]
 
-        ontology_importer = importers.IMFImporter(filepath=rules_file_path)
+        ontology_importer = importers.IMFImporter.from_file(rules_file_path)
         result = ImporterPipeline.try_verify(ontology_importer, role_enum)
 
         if result.rules is None:
@@ -387,7 +387,7 @@ class RulesInferenceFromRdfFile(Step):
         if role != "infer" and role is not None:
             role_enum = RoleTypes[role]
 
-        inference_importer = importers.InferenceImporter.from_rdf_file(
+        inference_importer = importers.InferenceImporter.from_file(
             rdf_file_path, max_number_of_instance=max_number_of_instance
         )
         result = ImporterPipeline.try_verify(inference_importer, role_enum)

cognite/neat/{workflows → _workflows}/steps/lib/current/rules_validator.py

@@ -5,14 +5,14 @@ from typing import ClassVar
 
 from cognite.client import CogniteClient
 
-from cognite.neat.issues import NeatIssueList
-from cognite.neat.issues.errors import ResourceNotFoundError, WorkflowStepNotInitializedError
-from cognite.neat.issues.formatters import FORMATTER_BY_NAME
-from cognite.neat.rules.models import DMSRules, SchemaCompleteness
-from cognite.neat.utils.cdf.loaders import ViewLoader
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import MultiRuleData
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._issues import NeatIssueList
+from cognite.neat._issues.errors import ResourceNotFoundError, WorkflowStepNotInitializedError
+from cognite.neat._issues.formatters import FORMATTER_BY_NAME
+from cognite.neat._rules.models import DMSRules, SchemaCompleteness
+from cognite.neat._utils.cdf.loaders import ViewLoader
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import MultiRuleData
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 CATEGORY = __name__.split(".")[-1].replace("_", " ").title()
 

cognite/neat/{workflows → _workflows}/steps/lib/io/io_steps.py

@@ -6,9 +6,9 @@ from typing import ClassVar
 import requests
 from cognite.client import CogniteClient
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 CATEGORY = "IO Steps"
 

cognite/neat/{workflows → _workflows}/steps/step_model.py

@@ -4,9 +4,9 @@ from typing import ClassVar, TypeVar
 
 from pydantic import BaseModel, ConfigDict
 
-from cognite.neat.app.monitoring.metrics import NeatMetricsCollector
-from cognite.neat.config import Config
-from cognite.neat.workflows.model import FlowMessage, WorkflowConfigs
+from cognite.neat._app.monitoring.metrics import NeatMetricsCollector
+from cognite.neat._config import Config
+from cognite.neat._workflows.model import FlowMessage, WorkflowConfigs
 
 
 class Configurable(BaseModel):

cognite/neat/{workflows → _workflows}/steps_registry.py

@@ -10,13 +10,13 @@ from typing import Any
 from pydantic import BaseModel
 
 # steps
-import cognite.neat.workflows.steps.lib.current
-import cognite.neat.workflows.steps.lib.io
-from cognite.neat.app.monitoring.metrics import NeatMetricsCollector
-from cognite.neat.config import Config
-from cognite.neat.issues.errors import WorkflowConfigurationNotSetError, WorkFlowMissingDataError
-from cognite.neat.workflows.model import FlowMessage, WorkflowConfigs
-from cognite.neat.workflows.steps.step_model import Configurable, DataContract, Step
+import cognite.neat._workflows.steps.lib.current
+import cognite.neat._workflows.steps.lib.io
+from cognite.neat._app.monitoring.metrics import NeatMetricsCollector
+from cognite.neat._config import Config
+from cognite.neat._issues.errors import WorkflowConfigurationNotSetError, WorkFlowMissingDataError
+from cognite.neat._workflows.model import FlowMessage, WorkflowConfigs
+from cognite.neat._workflows.steps.step_model import Configurable, DataContract, Step
 
 
 class StepMetadata(BaseModel):
@@ -50,12 +50,12 @@ class StepsRegistry:
         if self._step_classes:
             # classes already loaded - no need to reload
             return
-        for name, step_cls in inspect.getmembers(cognite.neat.workflows.steps.lib.current):
+        for name, step_cls in inspect.getmembers(cognite.neat._workflows.steps.lib.current):
             if inspect.isclass(step_cls):
                 logging.info(f"Loading NEAT step {name}")
                 self._step_classes.append(step_cls)
                 self.categorized_steps["current"].add(name)
-        for name, step_cls in inspect.getmembers(cognite.neat.workflows.steps.lib.io):
+        for name, step_cls in inspect.getmembers(cognite.neat._workflows.steps.lib.io):
             if inspect.isclass(step_cls):
                 logging.info(f"Loading NEAT step {name}")
                 self._step_classes.append(step_cls)