cognite-neat 0.128.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

cognite/neat/__init__.py

@@ -1,6 +1,7 @@
-from cognite.neat.v0.core._utils.auth import get_cognite_client
-from cognite.neat.v0.session import NeatSession
+from cognite.neat._v0.core._utils.auth import get_cognite_client
 
+from ._config import NeatConfig
+from ._session import NeatSession
 from ._version import __version__
 
-__all__ = ["NeatSession", "__version__", "get_cognite_client"]
+__all__ = ["NeatConfig", "NeatSession", "__version__", "get_cognite_client"]

cognite/neat/_config.py

@@ -1,9 +1,12 @@
 import sys
 from pathlib import Path
+from typing import Any, Literal, TypeAlias
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
+from cognite.neat._exceptions import UserInputError
 from cognite.neat._issues import ConsistencyError, ModelSyntaxError
+from cognite.neat._utils.text import humanize_collection
 from cognite.neat._utils.useful_types import ModusOperandi
 
 if sys.version_info >= (3, 11):
@@ -11,8 +14,14 @@ if sys.version_info >= (3, 11):
 else:
     import tomli  # type: ignore
 
+PredefinedProfile: TypeAlias = Literal["legacy-additive", "legacy-rebuild", "deep-additive", "deep-rebuild"]
 
-class ValidationConfig(BaseModel, populate_by_name=True):
+
+class ConfigModel(BaseModel):
+    model_config = ConfigDict(populate_by_name=True, validate_assignment=True)
+
+
+class ValidationConfig(ConfigModel):
     """Validation configuration."""
 
     exclude: list[str] = Field(default_factory=list)
@@ -71,18 +80,45 @@ class ValidationConfig(BaseModel, populate_by_name=True):
         return f"Excluded Rules: {', '.join(self.exclude)}"
 
 
-class ModelingConfig(BaseModel, populate_by_name=True):
+class ModelingConfig(ConfigModel):
     """Modeling configuration."""
 
     mode: ModusOperandi = "additive"
 
 
-class NeatConfig(BaseModel, populate_by_name=True):
+class AlphaFlagConfig(ConfigModel):
+    """Alpha feature flags configuration."""
+
+    fix_validation_issues: bool = Field(
+        default=False,
+        description="If enabled, Neat will attempt to automatically fix certain validation issues in the data model.",
+    )
+
+    def __setattr__(self, key: str, value: Any) -> None:
+        """Set attribute value or raise AttributeError."""
+        if key in self.model_fields:
+            super().__setattr__(key, value)
+        else:
+            available_flags = humanize_collection(type(self).model_fields.keys())
+            raise UserInputError(f"Alpha flag '{key}' not found. Available flags: {available_flags}.")
+
+    def _repr_html_(self) -> str:
+        """HTML representation for Jupyter notebooks."""
+        lines = ["<b>Alpha Feature Flags:</b>"]
+        for field_name, field in type(self).model_fields.items():
+            value = getattr(self, field_name)
+            display = "Enabled" if value else "Disabled"
+            lines.append(f"<li><b>{field_name}</b>: {display} - {field.description}</li>")
+        return "<ul>" + "\n".join(lines) + "</ul>"
+
+
+class NeatConfig(ConfigModel):
     """Configuration for a custom profile."""
 
     profile: str
     validation: ValidationConfig
     modeling: ModelingConfig
+    alpha: AlphaFlagConfig = Field(default_factory=AlphaFlagConfig)
 
     def __str__(self) -> str:
         """Human-readable configuration summary."""
@@ -93,8 +129,19 @@ class NeatConfig(BaseModel, populate_by_name=True):
         ]
         return "\n".join(lines)
 
+    @classmethod
+    def create_predefined(cls, profile: PredefinedProfile = "legacy-additive") -> "NeatConfig":
+        """Create NeatConfig from internal profiles."""
+        available_profiles = internal_profiles()
+        if profile not in available_profiles:
+            raise UserInputError(
+                f"Profile {profile!r} not found. Available predefined profiles: "
+                f"{humanize_collection(available_profiles.keys())}."
+            )
+        return available_profiles[profile]
+
 
-def internal_profiles() -> dict[str, NeatConfig]:
+def internal_profiles() -> dict[PredefinedProfile, NeatConfig]:
     """Get internal NeatConfig profile by name."""
     return {
         "legacy-additive": NeatConfig(
@@ -136,7 +183,7 @@ def internal_profiles() -> dict[str, NeatConfig]:
     }
 
 
-def get_neat_config(config_file_name: str, profile: str) -> NeatConfig:
+def get_neat_config_from_file(config_file_name: str, profile: str) -> NeatConfig:
     """Get NeatConfig from file or internal profiles.
 
     Args:

cognite/neat/_data_model/_identifiers.py

@@ -1,6 +1,6 @@
 from pydantic import HttpUrl, RootModel, ValidationError
 
-from cognite.neat.v0.core._utils.auxiliary import local_import
+from cognite.neat._v0.core._utils.auxiliary import local_import
 
 
 class URI(RootModel[str]):

cognite/neat/_data_model/importers/_table_importer/data_classes.py

@@ -18,7 +18,7 @@ from traitlets import Any
 from cognite.neat._data_model.models.entities import ParsedEntity, parse_entities, parse_entity
 from cognite.neat._utils.text import title_case
 from cognite.neat._utils.useful_types import CellValueType
-from cognite.neat.v0.core._data_model.models.entities import (
+from cognite.neat._v0.core._data_model.models.entities import (
     HasDataFilter,
     NodeTypeFilter,
     RawFilter,

cognite/neat/_data_model/models/conceptual/_data_model.py

@@ -4,7 +4,7 @@ from pydantic import Field, ValidationInfo, field_validator
 
 from cognite.neat._data_model.models.entities._constants import PREFIX_PATTERN, SUFFIX_PATTERN, VERSION_PATTERN
 from cognite.neat._utils.text import humanize_collection
-from cognite.neat.v0.core._data_model.models.entities._single_value import ConceptEntity
+from cognite.neat._v0.core._data_model.models.entities._single_value import ConceptEntity
 
 from ._base import ResourceMetadata
 from ._concept import Concept

cognite/neat/_data_model/models/dms/_space.py

@@ -3,7 +3,7 @@ from abc import ABC
 from pydantic import Field, field_validator
 
 from cognite.neat._data_model.models.dms._references import SpaceReference
-from cognite.neat.v0.core._utils.text import humanize_collection
+from cognite.neat._v0.core._utils.text import humanize_collection
 
 from ._base import APIResource, Resource, WriteableResource
 from ._constants import FORBIDDEN_SPACES, SPACE_FORMAT_PATTERN

cognite/neat/_data_model/models/entities/_identifiers.py

@@ -1,6 +1,6 @@
 from pydantic import HttpUrl, RootModel, ValidationError
 
-from cognite.neat.v0.core._utils.auxiliary import local_import
+from cognite.neat._v0.core._utils.auxiliary import local_import
 
 
 class URI(RootModel[str]):

cognite/neat/_session/_session.py

@@ -1,11 +1,10 @@
 import json
-from typing import Literal
 
 from cognite.client import ClientConfig, CogniteClient
 
 from cognite.neat import _version
 from cognite.neat._client import NeatClient
-from cognite.neat._config import internal_profiles
+from cognite.neat._config import NeatConfig, PredefinedProfile
 from cognite.neat._state_machine import EmptyState, PhysicalState
 from cognite.neat._store import NeatStore
 from cognite.neat._utils.http_client import ParametersRequest, SuccessResponse
@@ -20,24 +19,18 @@ class NeatSession:
     """A session is an interface for neat operations."""
 
     def __init__(
-        self,
-        client: CogniteClient | ClientConfig,
-        config: Literal["legacy-additive", "legacy-rebuild", "deep-additive", "deep-rebuild"] | None = None,
+        self, client: CogniteClient | ClientConfig, config: PredefinedProfile | NeatConfig = "legacy-additive"
     ) -> None:
         """Initialize a Neat session.
 
         Args:
             client (CogniteClient | ClientConfig): The Cognite client or client configuration to use for the session.
-            config (Literal["legacy-additive", "legacy-rebuild", "deep-additive", "deep-rebuild"] | None):
+            config (Literal["legacy-additive", "legacy-rebuild", "deep-additive", "deep-rebuild"] | NeatConfig):
                 The configuration profile to use for the session.
-                If None, the default profile "legacy-additive" is used. Meaning that Neat will perform additive modeling
-                and apply only validations that were part of the legacy Neat version."""
-
-        # Load configuration
-        if config and config not in internal_profiles():
-            raise ValueError(f"Profile '{config}' not found among internal profiles.")
-
-        self._config = internal_profiles()[config or "legacy-additive"]
+                Defaults to "legacy-additive". This means Neat will perform additive modeling
+                and apply only validations that were part of the legacy Neat version.
+        """
+        self._config = NeatConfig.create_predefined(config) if isinstance(config, str) else config
 
         # Use configuration for physical data model
         self._store = NeatStore()

cognite/neat/{v0 → _v0}/core/_client/_api/data_modeling_loaders.py

@@ -55,14 +55,14 @@ from cognite.client.data_classes.data_modeling.views import (
 from cognite.client.exceptions import CogniteAPIError
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite.neat.v0.core._client.data_classes.data_modeling import Component
-from cognite.neat.v0.core._client.data_classes.schema import DMSSchema
-from cognite.neat.v0.core._issues.warnings import CDFMaxIterationsWarning
-from cognite.neat.v0.core._shared import T_ID
-from cognite.neat.v0.core._utils.tarjan import tarjan
+from cognite.neat._v0.core._client.data_classes.data_modeling import Component
+from cognite.neat._v0.core._client.data_classes.schema import DMSSchema
+from cognite.neat._v0.core._issues.warnings import CDFMaxIterationsWarning
+from cognite.neat._v0.core._shared import T_ID
+from cognite.neat._v0.core._utils.tarjan import tarjan
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._client._api_client import NeatClient
+    from cognite.neat._v0.core._client._api_client import NeatClient
 
 T_WritableCogniteResourceList = TypeVar("T_WritableCogniteResourceList", bound=WriteableCogniteResourceList)
 

cognite/neat/{v0 → _v0}/core/_client/_api/neat_instances.py

@@ -7,13 +7,13 @@ from cognite.client.data_classes.filters import Filter
 from cognite.client.exceptions import CogniteAPIError
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite.neat.v0.core._constants import DMS_INSTANCE_LIMIT_MARGIN
-from cognite.neat.v0.core._issues import NeatIssue
-from cognite.neat.v0.core._issues.errors import WillExceedLimitError
-from cognite.neat.v0.core._issues.warnings import NeatValueWarning
+from cognite.neat._v0.core._constants import DMS_INSTANCE_LIMIT_MARGIN
+from cognite.neat._v0.core._issues import NeatIssue
+from cognite.neat._v0.core._issues.errors import WillExceedLimitError
+from cognite.neat._v0.core._issues.warnings import NeatValueWarning
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._client._api_client import NeatClient
+    from cognite.neat._v0.core._client._api_client import NeatClient
 
 
 class NeatInstancesAPI:

cognite/neat/{v0 → _v0}/core/_client/_api/schema.py

@@ -5,17 +5,17 @@ from typing import TYPE_CHECKING
 
 from cognite.client import data_modeling as dm
 
-from cognite.neat.v0.core._client.data_classes.data_modeling import (
+from cognite.neat._v0.core._client.data_classes.data_modeling import (
     ContainerApplyDict,
     SpaceApplyDict,
     ViewApplyDict,
 )
-from cognite.neat.v0.core._client.data_classes.schema import DMSSchema
-from cognite.neat.v0.core._constants import is_hierarchy_property
-from cognite.neat.v0.core._issues.errors import NeatValueError
+from cognite.neat._v0.core._client.data_classes.schema import DMSSchema
+from cognite.neat._v0.core._constants import is_hierarchy_property
+from cognite.neat._v0.core._issues.errors import NeatValueError
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._client._api_client import NeatClient
+    from cognite.neat._v0.core._client._api_client import NeatClient
 
 
 class SchemaAPI:

cognite/neat/{v0 → _v0}/core/_client/_api/statistics.py

@@ -7,7 +7,7 @@ from cognite.client.config import ClientConfig
 from cognite.client.data_classes.data_modeling.ids import _load_space_identifier
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite.neat.v0.core._client.data_classes.statistics import (
+from cognite.neat._v0.core._client.data_classes.statistics import (
     ProjectStatsAndLimits,
     SpaceInstanceCounts,
     SpaceInstanceCountsList,
@@ -34,7 +34,7 @@ class StatisticsAPI(APIClient):
         Examples:
             Fetch project statistics (and limits) and check the current number of data models vs.
             and how many more can be created:
-                >>> from cognite.neat.v0.core._client import NeatClient
+                >>> from cognite.neat._v0.core._client import NeatClient
                 >>> client = NeatClient()
                 >>> stats = client.instance_statistics.project()
                 >>> num_dm = stats.data_models.current
@@ -64,7 +64,7 @@ class StatisticsAPI(APIClient):
 
         Examples:
             Fetch statistics for a single space:
-                >>> from cognite.neat.v0.core._client import NeatClient
+                >>> from cognite.neat._v0.core._client import NeatClient
                 >>> client = NeatClient()
                 >>> res = client.instance_statistics.list("my-space")
             Fetch statistics for multiple spaces:

cognite/neat/{v0 → _v0}/core/_client/_api_client.py

@@ -1,6 +1,6 @@
 from cognite.client import ClientConfig, CogniteClient
 
-from cognite.neat.v0.core._utils.auth import _CLIENT_NAME
+from cognite.neat._v0.core._utils.auth import _CLIENT_NAME
 
 from ._api.data_modeling_loaders import DataModelLoaderAPI
 from ._api.neat_instances import NeatInstancesAPI

cognite/neat/{v0 → _v0}/core/_client/data_classes/schema.py

@@ -20,21 +20,21 @@ from cognite.client.data_classes.data_modeling.views import (
     ViewPropertyApply,
 )
 
-from cognite.neat.v0.core._client.data_classes.data_modeling import (
+from cognite.neat._v0.core._client.data_classes.data_modeling import (
     CogniteResourceDict,
     ContainerApplyDict,
     NodeApplyDict,
     SpaceApplyDict,
     ViewApplyDict,
 )
-from cognite.neat.v0.core._issues.errors import (
+from cognite.neat._v0.core._issues.errors import (
     NeatYamlError,
 )
-from cognite.neat.v0.core._issues.warnings import (
+from cognite.neat._v0.core._issues.warnings import (
     FileTypeUnexpectedWarning,
     ResourcesDuplicatedWarning,
 )
-from cognite.neat.v0.core._utils.text import to_camel_case
+from cognite.neat._v0.core._utils.text import to_camel_case
 
 if sys.version_info >= (3, 11):
     from typing import Self

cognite/neat/{v0 → _v0}/core/_client/testing.py

@@ -5,7 +5,7 @@ from unittest.mock import MagicMock
 
 from cognite.client.testing import CogniteClientMock
 
-from cognite.neat.v0.core._client._api_client import NeatClient
+from cognite.neat._v0.core._client._api_client import NeatClient
 
 from ._api.data_modeling_loaders import DataModelLoaderAPI
 from ._api.neat_instances import NeatInstancesAPI

cognite/neat/{v0 → _v0}/core/_constants.py

@@ -11,7 +11,7 @@ from rdflib.namespace import DefinedNamespace
 from cognite import neat
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._data_model.models.physical import PhysicalProperty
+    from cognite.neat._v0.core._data_model.models.physical import PhysicalProperty
 
 
 def _is_in_notebook() -> bool:
@@ -204,7 +204,7 @@ BASE_MODEL = Literal["CogniteCore"]
 
 def get_asset_read_only_properties_with_connection() -> "list[PhysicalProperty]":
     """Gets the asset read-only properties with connection, i.e. Root and Path."""
-    from cognite.neat.v0.core._data_model.models.physical import PhysicalProperty
+    from cognite.neat._v0.core._data_model.models.physical import PhysicalProperty
 
     return [PhysicalProperty.model_validate(item) for item in (_ASSET_ROOT_PROPERTY, _ASSET_PATH_PROPERTY)]
 
@@ -219,7 +219,7 @@ def get_base_concepts(
         total_concepts: The number of concepts to get. If None, all concepts are returned.
     """
     # Local import to avoid circular dependency issues
-    from cognite.neat.v0.core._issues.errors._general import NeatValueError
+    from cognite.neat._v0.core._issues.errors._general import NeatValueError
 
     if base_model == "CogniteCore":
         return [f"cdf_cdm:{concept}(version=v1)" for concept in COGNITE_CONCEPTS][:total_concepts]

cognite/neat/{v0 → _v0}/core/_data_model/_shared.py

@@ -1,15 +1,15 @@
 from dataclasses import dataclass
 from typing import Generic, TypeAlias, TypeVar
 
-from cognite.neat.v0.core._data_model.models import (
+from cognite.neat._v0.core._data_model.models import (
     ConceptualDataModel,
     PhysicalDataModel,
 )
-from cognite.neat.v0.core._data_model.models._import_contexts import ImportContext
-from cognite.neat.v0.core._data_model.models.conceptual._unverified import (
+from cognite.neat._v0.core._data_model.models._import_contexts import ImportContext
+from cognite.neat._v0.core._data_model.models.conceptual._unverified import (
     UnverifiedConceptualDataModel,
 )
-from cognite.neat.v0.core._data_model.models.physical._unverified import (
+from cognite.neat._v0.core._data_model.models.physical._unverified import (
     UnverifiedPhysicalDataModel,
 )
 

cognite/neat/{v0 → _v0}/core/_data_model/analysis/_base.py

@@ -11,23 +11,23 @@ import pandas as pd
 from cognite.client import data_modeling as dm
 from rdflib import URIRef
 
-from cognite.neat.v0.core._data_model.models import ConceptualDataModel, PhysicalDataModel
-from cognite.neat.v0.core._data_model.models.conceptual import (
+from cognite.neat._v0.core._data_model.models import ConceptualDataModel, PhysicalDataModel
+from cognite.neat._v0.core._data_model.models.conceptual import (
     Concept,
     ConceptualProperty,
 )
-from cognite.neat.v0.core._data_model.models.entities import (
+from cognite.neat._v0.core._data_model.models.entities import (
     ConceptEntity,
     MultiValueTypeInfo,
     ViewEntity,
 )
-from cognite.neat.v0.core._data_model.models.entities._single_value import (
+from cognite.neat._v0.core._data_model.models.entities._single_value import (
     UnknownEntity,
 )
-from cognite.neat.v0.core._data_model.models.physical import PhysicalProperty
-from cognite.neat.v0.core._data_model.models.physical._verified import PhysicalView
-from cognite.neat.v0.core._issues.errors import NeatValueError
-from cognite.neat.v0.core._issues.warnings import NeatValueWarning
+from cognite.neat._v0.core._data_model.models.physical import PhysicalProperty
+from cognite.neat._v0.core._data_model.models.physical._verified import PhysicalView
+from cognite.neat._v0.core._issues.errors import NeatValueError
+from cognite.neat._v0.core._issues.warnings import NeatValueWarning
 
 T_Hashable = TypeVar("T_Hashable", bound=Hashable)
 

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_base.py

@@ -5,14 +5,14 @@ from pathlib import Path
 from types import UnionType
 from typing import TYPE_CHECKING, Generic, TypeVar, Union, get_args, get_origin
 
-from cognite.neat.v0.core._client import NeatClient
-from cognite.neat.v0.core._constants import DEFAULT_NAMESPACE
-from cognite.neat.v0.core._data_model._shared import T_VerifiedDataModel
-from cognite.neat.v0.core._utils.auxiliary import class_html_doc
-from cognite.neat.v0.core._utils.upload import UploadResult, UploadResultList
+from cognite.neat._v0.core._client import NeatClient
+from cognite.neat._v0.core._constants import DEFAULT_NAMESPACE
+from cognite.neat._v0.core._data_model._shared import T_VerifiedDataModel
+from cognite.neat._v0.core._utils.auxiliary import class_html_doc
+from cognite.neat._v0.core._utils.upload import UploadResult, UploadResultList
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._store._provenance import Agent as ProvenanceAgent
+    from cognite.neat._v0.core._store._provenance import Agent as ProvenanceAgent
 
 T_Export = TypeVar("T_Export")
 
@@ -36,7 +36,7 @@ class BaseExporter(ABC, Generic[T_VerifiedDataModel, T_Export]):
     @property
     def agent(self) -> "ProvenanceAgent":
         """Provenance agent for the importer."""
-        from cognite.neat.v0.core._store._provenance import Agent as ProvenanceAgent
+        from cognite.neat._v0.core._store._provenance import Agent as ProvenanceAgent
 
         return ProvenanceAgent(id_=DEFAULT_NAMESPACE[f"agent/{type(self).__name__}"])
 

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_data_model2dms.py

@@ -18,24 +18,24 @@ from cognite.client.data_classes.data_modeling import (
 )
 from cognite.client.exceptions import CogniteAPIError
 
-from cognite.neat.v0.core._client import DataModelingLoader, NeatClient
-from cognite.neat.v0.core._client._api.data_modeling_loaders import (
+from cognite.neat._v0.core._client import DataModelingLoader, NeatClient
+from cognite.neat._v0.core._client._api.data_modeling_loaders import (
     MultiCogniteAPIError,
     T_WritableCogniteResourceList,
 )
-from cognite.neat.v0.core._client.data_classes.data_modeling import (
+from cognite.neat._v0.core._client.data_classes.data_modeling import (
     Component,
     ViewApplyDict,
 )
-from cognite.neat.v0.core._client.data_classes.schema import DMSSchema
-from cognite.neat.v0.core._data_model.models.physical import PhysicalDataModel
-from cognite.neat.v0.core._issues import IssueList
-from cognite.neat.v0.core._issues.warnings import (
+from cognite.neat._v0.core._client.data_classes.schema import DMSSchema
+from cognite.neat._v0.core._data_model.models.physical import PhysicalDataModel
+from cognite.neat._v0.core._issues import IssueList
+from cognite.neat._v0.core._issues.warnings import (
     PrincipleOneModelOneSpaceWarning,
     ResourceRetrievalWarning,
 )
-from cognite.neat.v0.core._shared import T_ID
-from cognite.neat.v0.core._utils.upload import UploadResult
+from cognite.neat._v0.core._shared import T_ID
+from cognite.neat._v0.core._utils.upload import UploadResult
 
 from ._base import CDFExporter
 

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_data_model2excel.py

@@ -14,24 +14,24 @@ from openpyxl.worksheet.datavalidation import DataValidation
 from openpyxl.worksheet.worksheet import Worksheet
 from rdflib import Namespace
 
-from cognite.neat.v0.core._constants import BASE_MODEL, get_base_concepts
-from cognite.neat.v0.core._data_model._constants import get_internal_properties
-from cognite.neat.v0.core._data_model._shared import VerifiedDataModel
-from cognite.neat.v0.core._data_model.models import (
+from cognite.neat._v0.core._constants import BASE_MODEL, get_base_concepts
+from cognite.neat._v0.core._data_model._constants import get_internal_properties
+from cognite.neat._v0.core._data_model._shared import VerifiedDataModel
+from cognite.neat._v0.core._data_model.models import (
     SheetRow,
 )
-from cognite.neat.v0.core._data_model.models._base_verified import (
+from cognite.neat._v0.core._data_model.models._base_verified import (
     BaseVerifiedMetadata,
     RoleTypes,
 )
-from cognite.neat.v0.core._data_model.models.conceptual._verified import (
+from cognite.neat._v0.core._data_model.models.conceptual._verified import (
     ConceptualDataModel,
 )
-from cognite.neat.v0.core._data_model.models.data_types import (
+from cognite.neat._v0.core._data_model.models.data_types import (
     _DATA_TYPE_BY_DMS_TYPE,
 )
-from cognite.neat.v0.core._data_model.models.physical._verified import PhysicalDataModel
-from cognite.neat.v0.core._utils.spreadsheet import (
+from cognite.neat._v0.core._data_model.models.physical._verified import PhysicalDataModel
+from cognite.neat._v0.core._utils.spreadsheet import (
     find_column_and_row_with_value,
     generate_data_validation,
 )

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_data_model2instance_template.py

@@ -8,12 +8,12 @@ from openpyxl.styles import Alignment, Border, Font, NamedStyle, PatternFill, Si
 from openpyxl.utils import get_column_letter
 from openpyxl.worksheet.datavalidation import DataValidation
 
-from cognite.neat.v0.core._data_model._constants import EntityTypes
-from cognite.neat.v0.core._data_model.analysis import DataModelAnalysis
-from cognite.neat.v0.core._data_model.models.conceptual._verified import (
+from cognite.neat._v0.core._data_model._constants import EntityTypes
+from cognite.neat._v0.core._data_model.analysis import DataModelAnalysis
+from cognite.neat._v0.core._data_model.models.conceptual._verified import (
     ConceptualDataModel,
 )
-from cognite.neat.v0.core._data_model.models.entities._single_value import ConceptEntity
+from cognite.neat._v0.core._data_model.models.entities._single_value import ConceptEntity
 
 from ._base import BaseExporter
 

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_data_model2semantic_model.py

@@ -8,22 +8,22 @@ from pydantic import BaseModel, ConfigDict, ValidationInfo, field_validator
 from rdflib import DCTERMS, OWL, RDF, RDFS, XSD, BNode, Graph, Literal, Namespace, URIRef
 from rdflib.collection import Collection as GraphCollection
 
-from cognite.neat.v0.core._constants import DEFAULT_NAMESPACE as NEAT_NAMESPACE
-from cognite.neat.v0.core._data_model._constants import EntityTypes
-from cognite.neat.v0.core._data_model.analysis import DataModelAnalysis
-from cognite.neat.v0.core._data_model.models.conceptual import (
+from cognite.neat._v0.core._constants import DEFAULT_NAMESPACE as NEAT_NAMESPACE
+from cognite.neat._v0.core._data_model._constants import EntityTypes
+from cognite.neat._v0.core._data_model.analysis import DataModelAnalysis
+from cognite.neat._v0.core._data_model.models.conceptual import (
     Concept,
     ConceptualDataModel,
     ConceptualMetadata,
     ConceptualProperty,
 )
-from cognite.neat.v0.core._data_model.models.data_types import DataType
-from cognite.neat.v0.core._data_model.models.entities import ConceptEntity
-from cognite.neat.v0.core._issues.errors import (
+from cognite.neat._v0.core._data_model.models.data_types import DataType
+from cognite.neat._v0.core._data_model.models.entities import ConceptEntity
+from cognite.neat._v0.core._issues.errors import (
     PropertyDefinitionDuplicatedError,
 )
-from cognite.neat.v0.core._issues.warnings import PropertyDefinitionDuplicatedWarning
-from cognite.neat.v0.core._utils.rdf_ import remove_namespace_from_uri
+from cognite.neat._v0.core._issues.warnings import PropertyDefinitionDuplicatedWarning
+from cognite.neat._v0.core._utils.rdf_ import remove_namespace_from_uri
 
 from ._base import BaseExporter
 

cognite/neat/{v0 → _v0}/core/_data_model/exporters/_data_model2yaml.py

@@ -5,7 +5,7 @@ from typing import Literal, get_args
 
 import yaml
 
-from cognite.neat.v0.core._data_model._shared import VerifiedDataModel
+from cognite.neat._v0.core._data_model._shared import VerifiedDataModel
 
 from ._base import BaseExporter
 

cognite/neat/{v0 → _v0}/core/_data_model/importers/_base.py

@@ -5,15 +5,15 @@ from typing import TYPE_CHECKING, Any, Generic
 
 from rdflib import URIRef
 
-from cognite.neat.v0.core._constants import DEFAULT_NAMESPACE
-from cognite.neat.v0.core._data_model._shared import (
+from cognite.neat._v0.core._constants import DEFAULT_NAMESPACE
+from cognite.neat._v0.core._data_model._shared import (
     ImportedDataModel,
     T_UnverifiedDataModel,
 )
-from cognite.neat.v0.core._utils.auxiliary import class_html_doc
+from cognite.neat._v0.core._utils.auxiliary import class_html_doc
 
 if TYPE_CHECKING:
-    from cognite.neat.v0.core._store._provenance import Agent as ProvenanceAgent
+    from cognite.neat._v0.core._store._provenance import Agent as ProvenanceAgent
 
 
 class BaseImporter(ABC, Generic[T_UnverifiedDataModel]):
@@ -51,7 +51,7 @@ class BaseImporter(ABC, Generic[T_UnverifiedDataModel]):
     @property
     def agent(self) -> "ProvenanceAgent":
         """Provenance agent for the importer."""
-        from cognite.neat.v0.core._store._provenance import Agent as ProvenanceAgent
+        from cognite.neat._v0.core._store._provenance import Agent as ProvenanceAgent
 
         return ProvenanceAgent(id_=DEFAULT_NAMESPACE[f"agent/{type(self).__name__}"])
 

cognite/neat/{v0 → _v0}/core/_data_model/importers/_base_file_reader.py

@@ -1,8 +1,8 @@
 from pathlib import Path
 from typing import Protocol, TypeVar
 
-from cognite.neat.v0.core._issues import NeatIssue
-from cognite.neat.v0.core._issues.errors import (
+from cognite.neat._v0.core._issues import NeatIssue
+from cognite.neat._v0.core._issues.errors import (
     FileNotAFileError,
     FileNotFoundNeatError,
     FileTypeUnexpectedError,