cognite-neat 0.123.32__py3-none-any.whl → 0.127.30__py3-none-any.whl

cognite/neat/v0/plugins/__init__.py

@@ -0,0 +1,4 @@
+from ._data_model import DataModelImporterPlugin, DataModelTransformerPlugin
+from ._manager import get_plugin_manager
+
+__all__ = ["DataModelImporterPlugin", "DataModelTransformerPlugin", "get_plugin_manager"]

cognite/neat/v0/plugins/_base.py

@@ -0,0 +1,9 @@
+from abc import ABC, abstractmethod
+from typing import Any
+
+
+class NeatPlugin(ABC):
+    @abstractmethod
+    def configure(self, *args: Any, **kwargs: Any) -> Any:
+        """A method that all plugins must implement."""
+        raise NotImplementedError()

cognite/neat/v0/plugins/_data_model.py

@@ -0,0 +1,48 @@
+from pathlib import Path
+from typing import Any
+
+from cognite.neat.v0.core._data_model.importers._base import BaseImporter
+from cognite.neat.v0.core._data_model.transformers._base import VerifiedDataModelTransformer
+
+from ._base import NeatPlugin
+
+
+class DataModelImporterPlugin(NeatPlugin):
+    """This class is used an interface for data model import plugins.
+    Any plugin that is used for importing data models should inherit from this class.
+    It is expected to implement the `configure` method which returns a configured importer.
+    """
+
+    def configure(self, io: str | Path | None, **kwargs: Any) -> BaseImporter:
+        """Return a configure plugin for data model import.
+        Args:
+            io (str | Path | None): The input/output interface for the plugin.
+            **kwargs (Any): Additional keyword arguments for configuration.
+        Returns:
+            BaseImporter: A configured instance of the BaseImporter.
+        !!! note "Returns"
+            The method must return an instance of `BaseImporter` or its subclasses
+            meaning it must implement the `BaseImporter` interface.
+        """
+
+        raise NotImplementedError()
+
+
+class DataModelTransformerPlugin(NeatPlugin):
+    """This class is used as an interface for data model transformer plugins.
+    Any plugin that is used for transforming data models should inherit from this class.
+    It is expected to implement the `configure` method which returns a configured transformer.
+    """
+
+    def configure(self, **kwargs: Any) -> VerifiedDataModelTransformer:
+        """Return a configure plugin for data model import.
+        Args:
+            **kwargs (Any): Additional keyword arguments for configuration.
+        Returns:
+            VerifiedDataModelTransformer: A configured instance of the VerifiedDataModelTransformer.
+        !!! note "Returns"
+            The method must return an instance of `VerifiedDataModelTransformer` or its subclasses
+            meaning it must implement the `VerifiedDataModelTransformer` interface.
+        """
+
+        raise NotImplementedError()

cognite/neat/{plugins → v0/plugins}/_issues.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from cognite.neat.core._issues.errors import _NEAT_ERRORS_BY_NAME, NeatError
+from cognite.neat.v0.core._issues.errors import _NEAT_ERRORS_BY_NAME, NeatError
 
 
 @dataclass(unsafe_hash=True)

cognite/neat/{plugins → v0/plugins}/_manager.py

@@ -1,21 +1,11 @@
 """Plugin manager for external plugins."""
 
 from importlib import metadata
-from typing import Any, ClassVar, TypeAlias, TypeVar, cast
+from typing import Any, ClassVar
 
+from ._base import NeatPlugin
+from ._data_model import DataModelImporterPlugin, DataModelTransformerPlugin
 from ._issues import PluginDuplicateError, PluginError, PluginLoadingError
-from .data_model.importers import DataModelImporterPlugin
-
-# Here we configure entry points where external plugins are going to be registered.
-plugins_entry_points = {
-    "cognite.neat.plugins.data_model.importers": DataModelImporterPlugin,
-}
-
-#: Type alias for all supported plugin types
-NeatPlugin: TypeAlias = DataModelImporterPlugin
-
-# Generic type variable for plugin types
-T_NeatPlugin = TypeVar("T_NeatPlugin", bound=NeatPlugin)
 
 
 class Plugin:
@@ -47,13 +37,14 @@ class PluginManager:
     """Plugin manager for external plugins."""
 
     _plugins_entry_points: ClassVar[dict[str, type[NeatPlugin]]] = {
-        "cognite.neat.plugins.data_model.importers": DataModelImporterPlugin,
+        "cognite.neat.v0.plugins.data_model.importers": DataModelImporterPlugin,
+        "cognite.neat.v0.plugins.data_model.transformers": DataModelTransformerPlugin,
     }
 
     def __init__(self, plugins: dict[tuple[str, type[NeatPlugin]], Any]) -> None:
         self._plugins = plugins
 
-    def get(self, name: str, type_: type[T_NeatPlugin]) -> type[T_NeatPlugin]:
+    def get(self, name: str, type_: type[NeatPlugin]) -> type[NeatPlugin]:
         """
         Returns desired plugin
 
@@ -63,7 +54,7 @@ class PluginManager:
         """
         try:
             plugin_class = self._plugins[(name, type_)]
-            return cast(type[T_NeatPlugin], plugin_class)
+            return plugin_class
         except KeyError:
             raise PluginError(plugin_name=name, plugin_type=type_.__name__) from None
 

cognite/neat/{session → v0/session}/_base.py

@@ -5,26 +5,27 @@ from cognite.client import CogniteClient
 from cognite.client import data_modeling as dm
 
 from cognite.neat import _version
-from cognite.neat.core._client import NeatClient
-from cognite.neat.core._data_model import importers
-from cognite.neat.core._data_model.models import PhysicalDataModel
-from cognite.neat.core._data_model.models.conceptual._verified import (
+from cognite.neat.v0.core._client import NeatClient
+from cognite.neat.v0.core._data_model import importers
+from cognite.neat.v0.core._data_model.models import PhysicalDataModel
+from cognite.neat.v0.core._data_model.models.conceptual._verified import (
     ConceptualDataModel,
 )
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._data_model.transformers import (
     ConceptualToPhysical,
     MergeConceptualDataModels,
     MergePhysicalDataModels,
     ToDMSCompliantEntities,
     VerifyConceptualDataModel,
 )
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._issues.errors import RegexViolationError
-from cognite.neat.core._issues.errors._general import NeatImportError
-from cognite.neat.core._store._data_model import DataModelEntity
-from cognite.neat.core._utils.auxiliary import local_import
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._issues.errors import RegexViolationError
+from cognite.neat.v0.core._issues.errors._general import NeatImportError
+from cognite.neat.v0.core._store._data_model import DataModelEntity
+from cognite.neat.v0.core._utils.auxiliary import local_import
 
 from ._collector import _COLLECTOR, Collector
+from ._diff import DiffAPI
 from ._drop import DropAPI
 from ._explore import ExploreAPI
 from ._fix import FixAPI
@@ -110,6 +111,7 @@ class NeatSession:
         self.template = TemplateAPI(self._state)
         self._explore = ExploreAPI(self._state)
         self.plugins = PluginAPI(self._state)
+        self._diff = DiffAPI(self._state)
         self.opt = OptAPI()
         self.opt._display()
         if load_engine != "skip" and (engine_version := load_neat_engine(client, load_engine)):

cognite/neat/{session → v0/session}/_collector.py

@@ -11,7 +11,7 @@ from typing import Any
 from mixpanel import Consumer, Mixpanel  # type: ignore[import-untyped]
 
 from cognite.neat._version import __version__
-from cognite.neat.core._constants import IN_NOTEBOOK, IN_PYODIDE
+from cognite.neat.v0.core._constants import IN_NOTEBOOK, IN_PYODIDE
 
 _NEAT_MIXPANEL_TOKEN: str = "bd630ad149d19999df3989c3a3750c4f"
 

cognite/neat/v0/session/_diff.py

@@ -0,0 +1,51 @@
+from typing import cast
+
+from rdflib.query import ResultRow
+
+from cognite.neat.v0.core._constants import NAMED_GRAPH_NAMESPACE
+
+from ._state import SessionState
+
+
+class DiffAPI:
+    """Compare RDF graphs (private API)."""
+
+    def __init__(self, state: SessionState) -> None:
+        self._state = state
+
+    def instances(self, current_named_graph: str, new_named_graph: str) -> None:
+        """
+        Compare two named graphs and store diff results.
+
+        Results stored in DIFF_ADD and DIFF_DELETE named graphs.
+
+        Args:
+            current_named_graph: Name of the current graph (e.g., "CURRENT")
+            new_named_graph: Name of the new graph (e.g., "NEW")
+        """
+        current_uri = NAMED_GRAPH_NAMESPACE[current_named_graph]
+        new_uri = NAMED_GRAPH_NAMESPACE[new_named_graph]
+
+        self._state.instances.store.diff(current_uri, new_uri)
+        self._print_summary()
+
+    def _print_summary(self) -> None:
+        """Print diff summary with triple counts."""
+        store = self._state.instances.store
+
+        add_query = (
+            f"SELECT (COUNT(*) as ?count) WHERE {{ GRAPH <{NAMED_GRAPH_NAMESPACE['DIFF_ADD']}> {{ ?s ?p ?o }} }}"
+        )
+        delete_query = (
+            f"SELECT (COUNT(*) as ?count) WHERE {{ GRAPH <{NAMED_GRAPH_NAMESPACE['DIFF_DELETE']}> {{ ?s ?p ?o }} }}"
+        )
+
+        add_result = cast(ResultRow, next(iter(store.dataset.query(add_query))))
+        delete_result = cast(ResultRow, next(iter(store.dataset.query(delete_query))))
+
+        add_count = int(add_result[0])
+        delete_count = int(delete_result[0])
+
+        print("Diff complete:")
+        print(f"  {add_count} triples to add (stored in DIFF_ADD)")
+        print(f"  {delete_count} triples to delete (stored in DIFF_DELETE)")

cognite/neat/{session → v0/session}/_drop.py

@@ -5,9 +5,9 @@ from typing import Literal
 from cognite.client.utils.useful_types import SequenceNotStr
 from rdflib import URIRef
 
-from cognite.neat.core._constants import COGNITE_MODELS
-from cognite.neat.core._data_model.transformers import DropModelViews
-from cognite.neat.core._issues import IssueList
+from cognite.neat.v0.core._constants import COGNITE_MODELS
+from cognite.neat.v0.core._data_model.transformers import DropModelViews
+from cognite.neat.v0.core._issues import IssueList
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_explore.py

@@ -3,8 +3,8 @@ from typing import cast
 import pandas as pd
 from rdflib import URIRef
 
-from cognite.neat.core._utils.rdf_ import remove_namespace_from_uri
-from cognite.neat.core._utils.text import humanize_collection
+from cognite.neat.v0.core._utils.rdf_ import remove_namespace_from_uri
+from cognite.neat.v0.core._utils.text import humanize_collection
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_fix.py

@@ -1,7 +1,7 @@
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._data_model.transformers import (
     ToCompliantEntities,
 )
-from cognite.neat.core._issues._base import IssueList
+from cognite.neat.v0.core._issues._base import IssueList
 
 from ._state import SessionState
 from .exceptions import session_class_wrapper

cognite/neat/{session → v0/session}/_inspect.py

@@ -4,9 +4,9 @@ from typing import Literal, overload
 
 import pandas as pd
 
-from cognite.neat.core._constants import IN_NOTEBOOK
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._utils.upload import (
+from cognite.neat.v0.core._constants import IN_NOTEBOOK
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._utils.upload import (
     UploadResult,
     UploadResultCore,
     UploadResultList,

cognite/neat/{session → v0/session}/_mapping.py

@@ -1,12 +1,12 @@
-from cognite.neat.core._data_model.models.mapping import load_classic_to_core_mapping
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._data_model.models.mapping import load_classic_to_core_mapping
+from cognite.neat.v0.core._data_model.transformers import (
     AsParentPropertyId,
     ChangeViewPrefix,
     IncludeReferenced,
     PhysicalDataModelMapper,
     VerifiedDataModelTransformer,
 )
-from cognite.neat.core._issues import IssueList
+from cognite.neat.v0.core._issues import IssueList
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_plugin.py

@@ -2,11 +2,10 @@ import warnings
 from pathlib import Path
 from typing import Any
 
-from cognite.neat.core._issues._base import IssueList
-from cognite.neat.core._utils.reader._base import NeatReader
-from cognite.neat.plugins import get_plugin_manager
-from cognite.neat.plugins.data_model.importers._base import DataModelImporterPlugin
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._issues._base import IssueList
+from cognite.neat.v0.core._utils.reader._base import NeatReader
+from cognite.neat.v0.plugins import DataModelImporterPlugin, get_plugin_manager
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from .exceptions import session_class_wrapper

cognite/neat/{session → v0/session}/_prepare.py

@@ -4,19 +4,19 @@ from typing import Any
 
 from rdflib import URIRef
 
-from cognite.neat.core._data_model.transformers import PrefixEntities, StandardizeNaming
-from cognite.neat.core._data_model.transformers._converters import StandardizeSpaceAndVersion
-from cognite.neat.core._instances.transformers import (
+from cognite.neat.v0.core._data_model.transformers import PrefixEntities, StandardizeNaming
+from cognite.neat.v0.core._data_model.transformers._converters import StandardizeSpaceAndVersion
+from cognite.neat.v0.core._instances.transformers import (
     ConnectionToLiteral,
     ConvertLiteral,
     LiteralToEntity,
     RelationshipAsEdgeTransformer,
 )
-from cognite.neat.core._instances.transformers._rdfpath import MakeConnectionOnExactMatch
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._issues.errors import NeatValueError
-from cognite.neat.core._utils.text import humanize_collection
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._instances.transformers._rdfpath import MakeConnectionOnExactMatch
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._issues.errors import NeatValueError
+from cognite.neat.v0.core._utils.text import humanize_collection
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_read.py

@@ -1,41 +1,45 @@
+import re
 import warnings
 from typing import Any, Literal, cast
 
 from cognite.client.data_classes.data_modeling import DataModelId, DataModelIdentifier
 from cognite.client.utils.useful_types import SequenceNotStr
+from rdflib import URIRef
 
-from cognite.neat.core._client import NeatClient
-from cognite.neat.core._constants import (
+from cognite.neat.v0.core._client import NeatClient
+from cognite.neat.v0.core._constants import (
     CLASSIC_CDF_NAMESPACE,
+    NAMED_GRAPH_NAMESPACE,
     get_default_prefixes_and_namespaces,
 )
-from cognite.neat.core._data_model import catalog, importers
-from cognite.neat.core._data_model.importers import BaseImporter
-from cognite.neat.core._data_model.models.entities._single_value import ViewEntity
-from cognite.neat.core._data_model.transformers import ClassicPrepareCore
-from cognite.neat.core._data_model.transformers._converters import (
+from cognite.neat.v0.core._data_model import catalog, importers
+from cognite.neat.v0.core._data_model._constants import SPACE_COMPLIANCE_REGEX
+from cognite.neat.v0.core._data_model.importers import BaseImporter
+from cognite.neat.v0.core._data_model.models.entities._single_value import ViewEntity
+from cognite.neat.v0.core._data_model.transformers import ClassicPrepareCore
+from cognite.neat.v0.core._data_model.transformers._converters import (
     ToEnterpriseModel,
     _SubsetEditableCDMPhysicalDataModel,
 )
-from cognite.neat.core._instances import examples as instances_examples
-from cognite.neat.core._instances import extractors
-from cognite.neat.core._instances.extractors._classic_cdf._base import InstanceIdPrefix
-from cognite.neat.core._instances.transformers import (
+from cognite.neat.v0.core._instances import examples as instances_examples
+from cognite.neat.v0.core._instances import extractors
+from cognite.neat.v0.core._instances.extractors._classic_cdf._base import InstanceIdPrefix
+from cognite.neat.v0.core._instances.transformers import (
     ConvertLiteral,
     LiteralToEntity,
     Transformers,
 )
-from cognite.neat.core._instances.transformers._prune_graph import (
+from cognite.neat.v0.core._instances.transformers._prune_graph import (
     AttachPropertyFromTargetToSource,
     PruneDeadEndEdges,
     PruneInstancesOfUnknownType,
     PruneTypes,
 )
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._issues.errors import NeatValueError
-from cognite.neat.core._issues.warnings import MissingCogniteClientWarning
-from cognite.neat.core._utils.reader import NeatReader
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._issues.errors import NeatValueError
+from cognite.neat.v0.core._issues.warnings import MissingCogniteClientWarning
+from cognite.neat.v0.core._utils.reader import NeatReader
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from ._wizard import NeatObjectType, RDFFileType, XMLFileType, object_wizard, rdf_dm_wizard, xml_format_wizard
@@ -826,13 +830,21 @@ class RDFReadAPI(BaseReadAPI):
         importer = importers.OWLImporter.from_file(reader.materialize_path(), source_name=f"file {reader!s}")
         return self._state.data_model_import(importer)
 
-    def instances(self, io: Any) -> IssueList:
+    def instances(self, io: Any, named_graph: str | None = None) -> IssueList:
         self._state._raise_exception_if_condition_not_met(
             "Read RDF Instances",
             empty_data_model_store_required=True,
         )
         reader = NeatReader.create(io)
-        self._state.instances.store.write(extractors.RdfFileExtractor(reader.materialize_path()))
+
+        # validate and convert named_graph to URI
+        named_graph_uri: URIRef | None = None
+        if named_graph:
+            if not re.match(SPACE_COMPLIANCE_REGEX, named_graph):
+                raise NeatValueError(f"Named graph '{named_graph}' does not comply with naming requirements. ")
+            named_graph_uri = NAMED_GRAPH_NAMESPACE[named_graph]
+
+        self._state.instances.store.write(extractors.RdfFileExtractor(reader.materialize_path()), named_graph_uri)
         return IssueList()
 
     def __call__(
@@ -840,6 +852,7 @@ class RDFReadAPI(BaseReadAPI):
         io: Any,
         type: NeatObjectType | None = None,
         source: RDFFileType | None = None,
+        named_graph: str | URIRef | None = None,
     ) -> IssueList:
         if type is None:
             type = object_wizard()
@@ -858,7 +871,7 @@ class RDFReadAPI(BaseReadAPI):
                 raise ValueError(f"Expected ontology, imf types or instances, got {source}")
 
         elif type == "instances":
-            return self.instances(io)
+            return self.instances(io, named_graph=named_graph)
 
         else:
             raise NeatSessionError(f"Expected data model or instances, got {type}")

cognite/neat/{session → v0/session}/_set.py

@@ -1,14 +1,14 @@
 from cognite.client import CogniteClient
 from cognite.client import data_modeling as dm
 
-from cognite.neat.core._client import NeatClient
-from cognite.neat.core._constants import COGNITE_MODELS
-from cognite.neat.core._data_model.models import PhysicalDataModel
-from cognite.neat.core._data_model.transformers import SetIDDMSModel
-from cognite.neat.core._instances.transformers import SetType
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._issues.errors import NeatValueError
-from cognite.neat.core._utils.text import humanize_collection
+from cognite.neat.v0.core._client import NeatClient
+from cognite.neat.v0.core._constants import COGNITE_MODELS
+from cognite.neat.v0.core._data_model.models import PhysicalDataModel
+from cognite.neat.v0.core._data_model.transformers import SetIDDMSModel
+from cognite.neat.v0.core._instances.transformers import SetType
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._issues.errors import NeatValueError
+from cognite.neat.v0.core._utils.text import humanize_collection
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_show.py

@@ -6,11 +6,11 @@ import networkx as nx
 from IPython.display import HTML, display
 from pyvis.network import Network as PyVisNetwork  # type: ignore
 
-from cognite.neat.core._constants import IN_NOTEBOOK, IN_PYODIDE
-from cognite.neat.core._data_model.analysis._base import DataModelAnalysis
-from cognite.neat.core._utils.io_ import to_directory_compatible
-from cognite.neat.core._utils.rdf_ import remove_namespace_from_uri, uri_display_name
-from cognite.neat.session.exceptions import NeatSessionError
+from cognite.neat.v0.core._constants import IN_NOTEBOOK, IN_PYODIDE
+from cognite.neat.v0.core._data_model.analysis._base import DataModelAnalysis
+from cognite.neat.v0.core._utils.io_ import to_directory_compatible
+from cognite.neat.v0.core._utils.rdf_ import remove_namespace_from_uri, uri_display_name
+from cognite.neat.v0.session.exceptions import NeatSessionError
 
 from ._state import SessionState
 from .exceptions import session_class_wrapper

cognite/neat/{session → v0/session}/_state.py

@@ -3,18 +3,18 @@ from typing import Literal, cast
 
 from rdflib import URIRef
 
-from cognite.neat.core._client import NeatClient
-from cognite.neat.core._data_model.importers import BaseImporter, InferenceImporter
-from cognite.neat.core._data_model.models import ConceptualDataModel, PhysicalDataModel
-from cognite.neat.core._data_model.models.conceptual._validation import ConceptualValidation
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._client import NeatClient
+from cognite.neat.v0.core._data_model.importers import BaseImporter, InferenceImporter
+from cognite.neat.v0.core._data_model.models import ConceptualDataModel, PhysicalDataModel
+from cognite.neat.v0.core._data_model.models.conceptual._validation import ConceptualValidation
+from cognite.neat.v0.core._data_model.transformers import (
     VerifiedDataModelTransformer,
 )
-from cognite.neat.core._instances.extractors import KnowledgeGraphExtractor
-from cognite.neat.core._issues import IssueList
-from cognite.neat.core._issues.warnings._models import ConversionToPhysicalModelImpossibleWarning
-from cognite.neat.core._store import NeatDataModelStore, NeatInstanceStore
-from cognite.neat.core._utils.upload import UploadResultList
+from cognite.neat.v0.core._instances.extractors import KnowledgeGraphExtractor
+from cognite.neat.v0.core._issues import IssueList
+from cognite.neat.v0.core._issues.warnings._models import ConversionToPhysicalModelImpossibleWarning
+from cognite.neat.v0.core._store import NeatDataModelStore, NeatInstanceStore
+from cognite.neat.v0.core._utils.upload import UploadResultList
 
 from .exceptions import NeatSessionError, _session_method_wrapper
 

cognite/neat/{session → v0/session}/_subset.py

@@ -1,15 +1,15 @@
 import warnings
 
-from cognite.neat.core._data_model.models.entities._single_value import (
+from cognite.neat.v0.core._data_model.models.entities._single_value import (
     ConceptEntity,
     ViewEntity,
 )
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._data_model.transformers import (
     SubsetConceptualDataModel,
     SubsetPhysicalDataModel,
 )
-from cognite.neat.core._issues._base import IssueList
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._issues._base import IssueList
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_template.py

@@ -3,23 +3,23 @@ from typing import Any, Literal
 
 from cognite.client.data_classes.data_modeling import DataModelIdentifier
 
-from cognite.neat.core._constants import BASE_MODEL
-from cognite.neat.core._data_model._shared import ImportedDataModel
-from cognite.neat.core._data_model.exporters import ExcelExporter
-from cognite.neat.core._data_model.importers import ExcelImporter
-from cognite.neat.core._data_model.models import UnverifiedConceptualDataModel
-from cognite.neat.core._data_model.models._base_verified import RoleTypes
-from cognite.neat.core._data_model.models.physical import PhysicalValidation
-from cognite.neat.core._data_model.transformers import (
+from cognite.neat.v0.core._constants import BASE_MODEL
+from cognite.neat.v0.core._data_model._shared import ImportedDataModel
+from cognite.neat.v0.core._data_model.exporters import ExcelExporter
+from cognite.neat.v0.core._data_model.importers import ExcelImporter
+from cognite.neat.v0.core._data_model.models import UnverifiedConceptualDataModel
+from cognite.neat.v0.core._data_model.models._base_verified import RoleTypes
+from cognite.neat.v0.core._data_model.models.physical import PhysicalValidation
+from cognite.neat.v0.core._data_model.transformers import (
     AddCogniteProperties,
     IncludeReferenced,
     ToDataProductModel,
     ToEnterpriseModel,
     VerifiedDataModelTransformer,
 )
-from cognite.neat.core._issues import IssueList, catch_issues
-from cognite.neat.core._utils.reader import NeatReader, PathReader
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._issues import IssueList, catch_issues
+from cognite.neat.v0.core._utils.reader import NeatReader, PathReader
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_to.py

@@ -6,18 +6,18 @@ from typing import Any, Literal, cast, overload
 from cognite.client import data_modeling as dm
 from cognite.client.data_classes.data_modeling import DataModelIdentifier
 
-from cognite.neat.core._client._api_client import NeatClient
-from cognite.neat.core._constants import COGNITE_MODELS
-from cognite.neat.core._data_model import exporters
-from cognite.neat.core._data_model._constants import PATTERNS
-from cognite.neat.core._data_model._shared import VerifiedDataModel
-from cognite.neat.core._data_model.importers import DMSImporter
-from cognite.neat.core._data_model.models import ConceptualDataModel, PhysicalDataModel
-from cognite.neat.core._data_model.models.physical import PhysicalMetadata
-from cognite.neat.core._instances import loaders
-from cognite.neat.core._issues import IssueList, NeatIssue, catch_issues
-from cognite.neat.core._utils.upload import UploadResultList
-from cognite.neat.session._experimental import ExperimentalFlags
+from cognite.neat.v0.core._client._api_client import NeatClient
+from cognite.neat.v0.core._constants import COGNITE_MODELS
+from cognite.neat.v0.core._data_model import exporters
+from cognite.neat.v0.core._data_model._constants import PATTERNS
+from cognite.neat.v0.core._data_model._shared import VerifiedDataModel
+from cognite.neat.v0.core._data_model.importers import DMSImporter
+from cognite.neat.v0.core._data_model.models import ConceptualDataModel, PhysicalDataModel
+from cognite.neat.v0.core._data_model.models.physical import PhysicalMetadata
+from cognite.neat.v0.core._instances import loaders
+from cognite.neat.v0.core._issues import IssueList, NeatIssue, catch_issues
+from cognite.neat.v0.core._utils.upload import UploadResultList
+from cognite.neat.v0.session._experimental import ExperimentalFlags
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

cognite/neat/{session → v0/session}/_wizard.py

@@ -3,7 +3,7 @@ from typing import Literal, TypeVar, get_args
 
 from rich.prompt import IntPrompt, Prompt
 
-from cognite.neat.core._data_model._constants import PATTERNS
+from cognite.neat.v0.core._data_model._constants import PATTERNS
 
 RDFFileType = Literal["Ontology", "IMF Types", "Inference"]
 NeatObjectType = Literal["Data Model", "Instances"]

cognite/neat/{session → v0/session}/engine/_load.py

@@ -13,7 +13,7 @@ from packaging.version import Version
 from packaging.version import parse as parse_version
 
 from cognite.neat._version import __engine__
-from cognite.neat.core._issues.errors import NeatValueError
+from cognite.neat.v0.core._issues.errors import NeatValueError
 
 ENVIRONMENT_VARIABLE = "NEATENGINE"
 PACKAGE_NAME = "neatengine"