cognite-neat 0.127.27__py3-none-any.whl → 0.127.29__py3-none-any.whl

cognite/neat/_config.py

@@ -0,0 +1,185 @@
+import sys
+from pathlib import Path
+
+from pydantic import BaseModel, Field
+
+from cognite.neat._issues import ConsistencyError, ModelSyntaxError
+from cognite.neat._utils.useful_types import ModusOperandi
+
+if sys.version_info >= (3, 11):
+    import tomllib as tomli  # Python 3.11+
+else:
+    import tomli  # type: ignore
+
+
+class ValidationConfig(BaseModel, populate_by_name=True):
+    """Validation configuration."""
+
+    exclude: list[str] = Field(default_factory=list)
+
+    def can_run_validator(self, code: str, issue_type: type) -> bool:
+        """
+        Check if a specific validator should run.
+
+        Args:
+            code: Validation code (e.g., "NEAT-DMS-CONTAINER-001")
+            issue_type: Issue type (e.g., ModelSyntaxError, ConsistencyError, Recommendation)
+
+        Returns:
+            True if validator should run, False otherwise
+        """
+
+        is_excluded = self._is_excluded(code, self.exclude)
+
+        if issue_type in [ModelSyntaxError, ConsistencyError] and is_excluded:
+            print(f"Validator {code} was excluded however it is a critical validator and will still run.")
+            return True
+        else:
+            return not is_excluded
+
+    @classmethod
+    def _is_excluded(cls, code: str, patterns: list[str]) -> bool:
+        """Check if code matches any pattern (supports wildcards)."""
+        for pattern in patterns:
+            if "*" in pattern:
+                # Split both pattern and code by hyphens
+                pattern_parts = pattern.split("-")
+                code_parts = code.split("-")
+
+                # Pattern must have same or fewer parts than code
+                if len(pattern_parts) > len(code_parts):
+                    continue
+
+                # Check if all pattern parts match (allowing wildcards)
+                match = True
+                for p_part, c_part in zip(pattern_parts, code_parts, strict=False):
+                    if p_part != "*" and p_part != c_part:
+                        match = False
+                        break
+
+                if match:
+                    return True
+            elif code == pattern:
+                return True
+
+        return False
+
+    def __str__(self) -> str:
+        """Human-readable configuration summary."""
+        if not self.exclude:
+            return "All validators enabled"
+        return f"Excluded Rules: {', '.join(self.exclude)}"
+
+
+class ModelingConfig(BaseModel, populate_by_name=True):
+    """Modeling configuration."""
+
+    mode: ModusOperandi = "additive"
+
+
+class NeatConfig(BaseModel, populate_by_name=True):
+    """Configuration for a custom profile."""
+
+    profile: str
+    validation: ValidationConfig
+    modeling: ModelingConfig
+
+    def __str__(self) -> str:
+        """Human-readable configuration summary."""
+        lines = [
+            f"Profile: {self.profile}",
+            f"Modeling Mode: {self.modeling.mode}",
+            f"Validation: {self.validation}",
+        ]
+        return "\n".join(lines)
+
+
+def internal_profiles() -> dict[str, NeatConfig]:
+    """Get internal NeatConfig profile by name."""
+    return {
+        "legacy-additive": NeatConfig(
+            profile="legacy-additive",
+            modeling=ModelingConfig(mode="additive"),
+            validation=ValidationConfig(
+                exclude=[
+                    "NEAT-DMS-AI-READINESS-*",
+                    "NEAT-DMS-CONNECTIONS-002",
+                    "NEAT-DMS-CONNECTIONS-REVERSE-007",
+                    "NEAT-DMS-CONNECTIONS-REVERSE-008",
+                    "NEAT-DMS-CONSISTENCY-001",
+                ]
+            ),
+        ),
+        "legacy-rebuild": NeatConfig(
+            profile="legacy-rebuild",
+            modeling=ModelingConfig(mode="rebuild"),
+            validation=ValidationConfig(
+                exclude=[
+                    "NEAT-DMS-AI-READINESS-*",
+                    "NEAT-DMS-CONNECTIONS-002",
+                    "NEAT-DMS-CONNECTIONS-REVERSE-007",
+                    "NEAT-DMS-CONNECTIONS-REVERSE-008",
+                    "NEAT-DMS-CONSISTENCY-001",
+                ]
+            ),
+        ),
+        "deep-additive": NeatConfig(
+            profile="deep-additive",
+            modeling=ModelingConfig(mode="additive"),
+            validation=ValidationConfig(exclude=[]),
+        ),
+        "deep-rebuild": NeatConfig(
+            profile="deep-rebuild",
+            modeling=ModelingConfig(mode="rebuild"),
+            validation=ValidationConfig(exclude=[]),
+        ),
+    }
+
+
+def get_neat_config(config_file_name: str, profile: str) -> NeatConfig:
+    """Get NeatConfig from file or internal profiles.
+
+    Args:
+        config_file_name: Path to configuration file.
+        profile: Profile name to use.
+    Returns:
+        NeatConfig instance.
+    """
+
+    if not config_file_name.endswith(".toml"):
+        raise ValueError("config_file_name must end with '.toml'")
+
+    file_path = Path.cwd() / config_file_name
+
+    if file_path.exists():
+        with file_path.open("rb") as f:
+            toml = tomli.load(f)
+
+        if "tool" in toml and "neat" in toml["tool"]:
+            data = toml["tool"]["neat"]
+        elif "neat" in toml:
+            data = toml["neat"]
+        else:
+            raise ValueError("No [tool.neat] or [neat] section found in the configuration file.")
+
+        toml_profile = data.get("profile")
+        toml_profiles = data.get("profiles")
+        hardcoded_profiles = internal_profiles()
+
+        if toml_profile and toml_profile in hardcoded_profiles:
+            raise ValueError(f"Internal profile '{toml_profile}' cannot be used in external configuration file.")
+
+        if toml_profiles and any(p in hardcoded_profiles for p in toml_profiles.keys()):
+            raise ValueError(
+                "Internal profiles cannot be redefined in external configuration file: "
+                f"{set(hardcoded_profiles.keys()).intersection(toml_profiles.keys())}"
+            )
+
+        if toml_profile and profile == toml_profile:
+            return NeatConfig(**data)
+        elif (built_in_profiles := data.get("profiles")) and profile in built_in_profiles:
+            return NeatConfig(profile=profile, **data["profiles"][profile])
+        else:
+            raise ValueError(f"Profile '{profile}' not found in configuration file.")
+    else:
+        raise FileNotFoundError(f"Configuration file '{file_path}' not found.")

cognite/neat/_data_model/validation/dms/_ai_readiness.py

@@ -23,6 +23,7 @@ class DataModelMissingName(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-001"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -61,6 +62,7 @@ class DataModelMissingDescription(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-002"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -95,6 +97,7 @@ class ViewMissingName(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-003"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -145,6 +148,7 @@ class ViewMissingDescription(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-004"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -185,6 +189,7 @@ class ViewPropertyMissingName(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-005"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -233,6 +238,7 @@ class ViewPropertyMissingDescription(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-006"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []

cognite/neat/_data_model/validation/dms/_base.py

@@ -58,6 +58,7 @@ class DataModelValidator(ABC):
     """Assessors for fundamental data model principles."""
 
     code: ClassVar[str]
+    issue_type: ClassVar[type[ConsistencyError] | type[Recommendation]]
 
     def __init__(
         self,

cognite/neat/_data_model/validation/dms/_connections.py

@@ -31,6 +31,7 @@ class ConnectionValueTypeUnexisting(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-001"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         undefined_value_types = []
@@ -80,6 +81,7 @@ class ConnectionValueTypeUndefined(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-002"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         missing_value_types = []
@@ -149,6 +151,7 @@ class ReverseConnectionSourceViewMissing(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-001"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -193,6 +196,7 @@ class ReverseConnectionSourcePropertyMissing(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-002"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -240,6 +244,7 @@ class ReverseConnectionSourcePropertyWrongType(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-003"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -290,6 +295,7 @@ class ReverseConnectionContainerMissing(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-004"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -347,6 +353,7 @@ class ReverseConnectionContainerPropertyMissing(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-005"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -407,6 +414,7 @@ class ReverseConnectionContainerPropertyWrongType(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-006"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -469,6 +477,7 @@ class ReverseConnectionTargetMissing(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-007"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -525,6 +534,7 @@ class ReverseConnectionPointsToAncestor(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-008"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []
@@ -584,6 +594,7 @@ class ReverseConnectionTargetMismatch(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-REVERSE-009"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []

cognite/neat/_data_model/validation/dms/_consistency.py

@@ -23,6 +23,7 @@ class ViewSpaceVersionInconsistentWithDataModel(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-001"
+    issue_type = Recommendation
 
     def run(self) -> list[Recommendation]:
         recommendations: list[Recommendation] = []

cognite/neat/_data_model/validation/dms/_containers.py

@@ -31,6 +31,7 @@ class ExternalContainerDoesNotExist(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-001"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -81,6 +82,7 @@ class ExternalContainerPropertyDoesNotExist(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-002"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -137,6 +139,7 @@ class RequiredContainerDoesNotExist(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-003"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []

cognite/neat/_data_model/validation/dms/_limits.py

@@ -38,6 +38,7 @@ class DataModelViewCountIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-DATA-MODEL-001"
+    issue_type = ConsistencyError
 
     def __init__(
         self,
@@ -82,6 +83,7 @@ class ViewPropertyCountIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-VIEW-001"
+    issue_type = ConsistencyError
 
     def __init__(
         self,
@@ -145,6 +147,7 @@ class ViewContainerCountIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-VIEW-002"
+    issue_type = ConsistencyError
 
     def __init__(
         self,
@@ -204,6 +207,7 @@ class ViewImplementsCountIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-VIEW-003"
+    issue_type = ConsistencyError
 
     def __init__(
         self,
@@ -258,6 +262,7 @@ class ContainerPropertyCountIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-CONTAINER-001"
+    issue_type = ConsistencyError
 
     def __init__(
         self,
@@ -326,6 +331,7 @@ class ContainerPropertyListSizeIsOutOfLimits(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-CONTAINER-002"
+    issue_type = ConsistencyError
 
     def __init__(
         self,

cognite/neat/_data_model/validation/dms/_orchestrator.py

@@ -1,3 +1,4 @@
+from collections.abc import Callable
 from itertools import chain
 
 from cognite.neat._client import NeatClient
@@ -56,12 +57,12 @@ class DmsDataModelValidation(OnSuccessIssuesChecker):
     def __init__(
         self,
         client: NeatClient | None = None,
-        codes: list[str] | None = None,
         modus_operandi: ModusOperandi = "additive",
+        can_run_validator: Callable[[str, type], bool] | None = None,
     ) -> None:
         super().__init__()
         self._client = client
-        self._codes = codes or ["all"]
+        self._can_run_validator = can_run_validator or (lambda code, issue_type: True)  # type: ignore
         self._modus_operandi = modus_operandi
         self._has_run = False
 
@@ -155,7 +156,7 @@ class DmsDataModelValidation(OnSuccessIssuesChecker):
 
         # Run validators
         for validator in validators:
-            if "all" in self._codes or validator.code in self._codes:
+            if self._can_run_validator(validator.code, validator.issue_type):
                 self._issues.extend(validator.run())
 
         self._has_run = True

cognite/neat/_data_model/validation/dms/_views.py

@@ -24,6 +24,7 @@ class ViewToContainerMappingNotPossible(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-001"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []
@@ -81,6 +82,7 @@ class ImplementedViewNotExisting(DataModelValidator):
     """
 
     code = f"{BASE_CODE}-002"
+    issue_type = ConsistencyError
 
     def run(self) -> list[ConsistencyError]:
         errors: list[ConsistencyError] = []

cognite/neat/_session/_physical.py

@@ -1,6 +1,7 @@
 from typing import Any, Literal
 
 from cognite.neat._client import NeatClient
+from cognite.neat._config import NeatConfig
 from cognite.neat._data_model.deployer.deployer import DeploymentOptions, SchemaDeployer
 from cognite.neat._data_model.exporters import (
     DMSAPIExporter,
@@ -19,7 +20,6 @@ from cognite.neat._exceptions import UserInputError
 from cognite.neat._state_machine import PhysicalState
 from cognite.neat._store._store import NeatStore
 from cognite.neat._utils._reader import NeatReader
-from cognite.neat._utils.useful_types import ModusOperandi
 
 from ._wrappers import session_wrapper
 
@@ -27,11 +27,12 @@ from ._wrappers import session_wrapper
 class PhysicalDataModel:
     """Read from a data source into NeatSession graph store."""
 
-    def __init__(self, store: NeatStore, client: NeatClient, mode: ModusOperandi) -> None:
+    def __init__(self, store: NeatStore, client: NeatClient, config: NeatConfig) -> None:
         self._store = store
         self._client = client
-        self.read = ReadPhysicalDataModel(self._store, self._client)
-        self.write = WritePhysicalDataModel(self._store, self._client, mode)
+        self._config = config
+        self.read = ReadPhysicalDataModel(self._store, self._client, self._config)
+        self.write = WritePhysicalDataModel(self._store, self._client, self._config)
 
     def _repr_html_(self) -> str:
         if not isinstance(self._store.state, PhysicalState):
@@ -69,9 +70,10 @@ class PhysicalDataModel:
 class ReadPhysicalDataModel:
     """Read physical data model from various sources into NeatSession graph store."""
 
-    def __init__(self, store: NeatStore, client: NeatClient) -> None:
+    def __init__(self, store: NeatStore, client: NeatClient, config: NeatConfig) -> None:
         self._store = store
         self._client = client
+        self._config = config
 
     def yaml(self, io: Any, format: Literal["neat", "toolkit"] = "neat") -> None:
         """Read physical data model from YAML file(s)
@@ -92,7 +94,11 @@ class ReadPhysicalDataModel:
             reader = DMSAPIImporter.from_yaml(path)
         else:
             raise UserInputError(f"Unsupported format: {format}. Supported formats are 'neat' and 'toolkit'.")
-        on_success = DmsDataModelValidation(self._client)
+        on_success = DmsDataModelValidation(
+            self._client,
+            modus_operandi=self._config.modeling.mode,
+            can_run_validator=self._config.validation.can_run_validator,
+        )
 
         return self._store.read_physical(reader, on_success)
 
@@ -115,7 +121,11 @@ class ReadPhysicalDataModel:
             reader = DMSAPIImporter.from_json(path)
         else:
             raise UserInputError(f"Unsupported format: {format}. Supported formats are 'neat' and 'toolkit'.")
-        on_success = DmsDataModelValidation(self._client)
+        on_success = DmsDataModelValidation(
+            self._client,
+            modus_operandi=self._config.modeling.mode,
+            can_run_validator=self._config.validation.can_run_validator,
+        )
 
         return self._store.read_physical(reader, on_success)
 
@@ -124,7 +134,11 @@ class ReadPhysicalDataModel:
 
         path = NeatReader.create(io).materialize_path()
         reader = DMSTableImporter.from_excel(path)
-        on_success = DmsDataModelValidation(self._client)
+        on_success = DmsDataModelValidation(
+            self._client,
+            modus_operandi=self._config.modeling.mode,
+            can_run_validator=self._config.validation.can_run_validator,
+        )
 
         return self._store.read_physical(reader, on_success)
 
@@ -140,7 +154,11 @@ class ReadPhysicalDataModel:
         reader = DMSAPIImporter.from_cdf(
             DataModelReference(space=space, external_id=external_id, version=version), self._client
         )
-        on_success = DmsDataModelValidation(self._client)
+        on_success = DmsDataModelValidation(
+            self._client,
+            modus_operandi=self._config.modeling.mode,
+            can_run_validator=self._config.validation.can_run_validator,
+        )
 
         return self._store.read_physical(reader, on_success)
 
@@ -149,10 +167,10 @@ class ReadPhysicalDataModel:
 class WritePhysicalDataModel:
     """Write physical data model to various sources from NeatSession graph store."""
 
-    def __init__(self, store: NeatStore, client: NeatClient, mode: ModusOperandi) -> None:
+    def __init__(self, store: NeatStore, client: NeatClient, config: NeatConfig) -> None:
         self._store = store
         self._client = client
-        self._mode = mode
+        self._config = config
 
     def yaml(self, io: Any, format: Literal["neat", "toolkit"] = "neat") -> None:
         """Write physical data model to YAML file
@@ -215,7 +233,7 @@ class WritePhysicalDataModel:
         """Write physical data model with views, containers, and spaces that are in the same space as the data model
         to CDF.
 
-        This method depends on the session mode set when creating the NeatSession.
+        This method depends on the session governance profile for data modeling set when creating the NeatSession.
             - In 'additive' mode, only new or updates to data models/views/containers will be applied.
                 You cannot remove views from data models, properties from views or containers, or
                 indexes or constraints from containers.
@@ -234,7 +252,10 @@ class WritePhysicalDataModel:
         """
         writer = DMSAPIExporter()
         options = DeploymentOptions(
-            dry_run=dry_run, auto_rollback=rollback, drop_data=drop_data, modus_operandi=self._mode
+            dry_run=dry_run,
+            auto_rollback=rollback,
+            drop_data=drop_data,
+            modus_operandi=self._config.modeling.mode,
         )
         on_success = SchemaDeployer(self._client, options)
         return self._store.write_physical(writer, on_success)

cognite/neat/_session/_session.py

@@ -1,13 +1,14 @@
 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._state_machine import EmptyState, PhysicalState
 from cognite.neat._store import NeatStore
 from cognite.neat._utils.http_client import ParametersRequest, SuccessResponse
-from cognite.neat._utils.useful_types import ModusOperandi
 
 from ._issues import Issues
 from ._opt import Opt
@@ -16,21 +17,38 @@ from ._result import Result
 
 
 class NeatSession:
-    """A session is an interface for neat operations. It works as
-    a manager for handling user interactions and orchestrating
-    the state machine for data model and instance operations.
-    """
+    """A session is an interface for neat operations."""
 
-    def __init__(self, client: CogniteClient | ClientConfig, mode: ModusOperandi = "additive") -> None:
+    def __init__(
+        self,
+        client: CogniteClient | ClientConfig,
+        config: Literal["legacy-additive", "legacy-rebuild", "deep-additive", "deep-rebuild"] | None = None,
+    ) -> 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):
+                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"]
+
+        # Use configuration for physical data model
         self._store = NeatStore()
         self._client = NeatClient(client)
-        self.physical_data_model = PhysicalDataModel(self._store, self._client, mode)
+        self.physical_data_model = PhysicalDataModel(self._store, self._client, self._config)
         self.issues = Issues(self._store)
         self.result = Result(self._store)
         self.opt = Opt(self._store)
 
         if self.opt._collector.can_collect:
-            self.opt._collector.collect("initSession", {"mode": mode})
+            self.opt._collector.collect("initSession", {"mode": self._config.modeling.mode})
 
         self._welcome_message()
 
@@ -50,6 +68,7 @@ class NeatSession:
                 message += f" (Organization: '{organization}')"
 
         print(message)
+        print(self._config)
 
     @property
     def version(self) -> str:

cognite/neat/_version.py

@@ -1,2 +1,2 @@
-__version__ = "0.127.27"
+__version__ = "0.127.29"
 __engine__ = "^2.0.4"

cognite/neat/v0/core/_data_model/models/physical/_exporter.py

@@ -243,17 +243,21 @@ class _DMSExporter:
             connection = cast(EdgeEntity, prop.connection)
 
             if connection.direction == "inwards" and isinstance(prop.value_type, ViewEntity):
-                edge_type_candidates = outwards_type_by_view_value_type.get((prop.view, prop.value_type), [])
-                if len(edge_type_candidates) == 0:
-                    # Warning in validation, should not have an inwards connection without an outwards connection
-                    edge_type = cls._default_edge_type_from_view_id(prop.view.as_id(), prop_id)
-                elif len(edge_type_candidates) == 1:
-                    edge_type = edge_type_candidates[0]
+                if connection.edge_type is not None:
+                    edge_type = connection.edge_type.as_reference()
                 else:
-                    raise NeatValueError(
-                        f"Cannot infer edge type for {view_id}.{prop_id}, multiple candidates: {edge_type_candidates}."
-                        "Please specify edge type explicitly, i.e., edge(type=<YOUR_TYPE>)."
-                    )
+                    edge_type_candidates = outwards_type_by_view_value_type.get((prop.view, prop.value_type), [])
+                    if len(edge_type_candidates) == 0:
+                        # Warning in validation, should not have an inwards connection without an outwards connection
+                        edge_type = cls._default_edge_type_from_view_id(prop.view.as_id(), prop_id)
+                    elif len(edge_type_candidates) == 1:
+                        edge_type = edge_type_candidates[0]
+                    else:
+                        raise NeatValueError(
+                            f"Cannot infer edge type for {view_id}.{prop_id}, multiple candidates: "
+                            f"{edge_type_candidates}. "
+                            "Please specify edge type explicitly, i.e., edge(type=<YOUR_TYPE>)."
+                        )
                 view_property_id = (prop.view, prop.view_property)
                 edge_types_by_view_property_id[view_property_id] = edge_type
 

{cognite_neat-0.127.27.dist-info → cognite_neat-0.127.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognite-neat
-Version: 0.127.27
+Version: 0.127.29
 Summary: Knowledge graph transformation
 Project-URL: Documentation, https://cognite-neat.readthedocs-hosted.com/
 Project-URL: Homepage, https://cognite-neat.readthedocs-hosted.com/

{cognite_neat-0.127.27.dist-info → cognite_neat-0.127.29.dist-info}/RECORD

@@ -1,7 +1,8 @@
 cognite/neat/__init__.py,sha256=Lo4DbjDOwnhCYUoAgPp5RG1fDdF7OlnomalTe7n1ydw,211
+cognite/neat/_config.py,sha256=W42Lj5PNR9X1is9vIPHLHvkl_rpU5PCrDuQI4zSNTJQ,6359
 cognite/neat/_exceptions.py,sha256=ox-5hXpee4UJlPE7HpuEHV2C96aLbLKo-BhPDoOAzhA,1650
 cognite/neat/_issues.py,sha256=wH1mnkrpBsHUkQMGUHFLUIQWQlfJ_qMfdF7q0d9wNhY,1871
-cognite/neat/_version.py,sha256=cz0CG5j6gdpaAmogpF9j31GRp2vxv4fKDwo6HE72kx0,47
+cognite/neat/_version.py,sha256=HnkgU0oEwuWFxHjFpJFKDNKAV0Zn342c-qMWtztHfKE,47
 cognite/neat/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognite/neat/v1.py,sha256=owqW5Mml2DSZx1AvPvwNRTBngfhBNrQ6EH-7CKL7Jp0,61
 cognite/neat/_client/__init__.py,sha256=75Bh7eGhaN4sOt3ZcRzHl7pXaheu1z27kmTHeaI05vo,114
@@ -74,20 +75,20 @@ cognite/neat/_data_model/models/entities/_identifiers.py,sha256=uBiK4ot3V0b_LGXu
 cognite/neat/_data_model/models/entities/_parser.py,sha256=zef_pSDZYMZrJl4IKreFDR577KutfhtN1xpH3Ayjt2o,7669
 cognite/neat/_data_model/validation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognite/neat/_data_model/validation/dms/__init__.py,sha256=Ps26r8bY4sLPsceaS6CpgS-s6JxN_zKbiteF5RrQYYw,2512
-cognite/neat/_data_model/validation/dms/_ai_readiness.py,sha256=_7R4BVVcJw27OYmMS-GxxcvQdocr67sU1gGouUo81EU,10525
-cognite/neat/_data_model/validation/dms/_base.py,sha256=HJRoo1fX2Fu0bzlPKhn3VOsrtaOv4ea49BxvEpfOjIY,13272
-cognite/neat/_data_model/validation/dms/_connections.py,sha256=goIcsGV4yYb-d9ltXg4eM5qxt8iErIIxtkiKJXZUYyM,26937
-cognite/neat/_data_model/validation/dms/_consistency.py,sha256=K0mfAwwpW6zQXGDEldrA7Lgqk-giV_Y-gUKyb8Dw47k,2445
-cognite/neat/_data_model/validation/dms/_containers.py,sha256=Jqwu2JH36eyqA7xt6_UaLEl-zhZi4itI1dmb73GpJMY,7431
-cognite/neat/_data_model/validation/dms/_limits.py,sha256=LVxF1qmeEdUVVAPmywCgxWJ9BR7Ai6nI4X1ylJ7pv8I,15982
-cognite/neat/_data_model/validation/dms/_orchestrator.py,sha256=1cwGrtsSs6Q1vT-8vbZ2DM_3WYu9YTSRDbDGJhkTCBs,10672
-cognite/neat/_data_model/validation/dms/_views.py,sha256=3bHEEbFKTR_QH_tiJYHppQsZ9ruApv-kdyfehEjIlCU,4198
+cognite/neat/_data_model/validation/dms/_ai_readiness.py,sha256=yKT9KHS4iD7ijKBjjAGbREKWDGunBuDTl9FCX0RiW2Q,10717
+cognite/neat/_data_model/validation/dms/_base.py,sha256=3Lqs8pzMQZznxhUr8esEe9H7F2wM3EKEl2H64GLS5OA,13344
+cognite/neat/_data_model/validation/dms/_connections.py,sha256=6ea8WZNqYT9SPluwXvg0mgDJpzZMqpPC5xonbdLrT4E,27305
+cognite/neat/_data_model/validation/dms/_consistency.py,sha256=uJ6coAVupD3WfeeXxoCIebM8WSnR78GXBXIBnN59aao,2477
+cognite/neat/_data_model/validation/dms/_containers.py,sha256=UuvzrBBw45F5f9uzCh97lW4t7m7XIIT1I9FnzzYUYv4,7533
+cognite/neat/_data_model/validation/dms/_limits.py,sha256=UDJ3oY2Xp96Zw2QpsGM4MQ6gZCrDuoKhhuEY2uQJLP4,16186
+cognite/neat/_data_model/validation/dms/_orchestrator.py,sha256=0KpXMWlofpELCE3QXQh6SDyP3IoATPrR1412HJzAN6w,10811
+cognite/neat/_data_model/validation/dms/_views.py,sha256=M4egIa7UAMGtZlqzIxx6ZzL4e_qo8GbDGh7vs9wywD8,4266
 cognite/neat/_session/__init__.py,sha256=owqW5Mml2DSZx1AvPvwNRTBngfhBNrQ6EH-7CKL7Jp0,61
 cognite/neat/_session/_issues.py,sha256=E8UQeSJURg2dm4MF1pfD9dp-heSRT7pgQZgKlD1-FGs,2723
 cognite/neat/_session/_opt.py,sha256=QcVK08JMmVzJpD0GKHelbljMOQi6CMD1w-maQOlbyZQ,1350
-cognite/neat/_session/_physical.py,sha256=VyMvvPyN9khR_26rMS0kVuZg23k5c8Hfs7XDeMrkF_w,10526
+cognite/neat/_session/_physical.py,sha256=0EdpFS5v4pk4XJju8vZm3pHIM4AlSaCjxGRKeMAcq9g,11304
 cognite/neat/_session/_result.py,sha256=po2X4s-Tioe0GQAGCfK862hKXNRX5YjJZsEzNcTC8nI,7879
-cognite/neat/_session/_session.py,sha256=rezw1MEMN8DPA6waULt5-mdeMuwq3ZNnSb1CSsKM0ko,2584
+cognite/neat/_session/_session.py,sha256=tWEvzi_85Ji52i2q5vJ7xIg3yR-8ywVCH8FjgncdlnU,3471
 cognite/neat/_session/_wrappers.py,sha256=9t_MnJ0Sw_v-f6oTIh8dtAT-3oEbqumGuND97aPCC3M,3581
 cognite/neat/_session/_html/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognite/neat/_session/_html/_render.py,sha256=fD8iee4ql50CrHGH41SSh9Tw1lM0tHt-NF0OnnxHosg,1193
@@ -190,7 +191,7 @@ cognite/neat/v0/core/_data_model/models/mapping/__init__.py,sha256=T68Hf7rhiXa7b
 cognite/neat/v0/core/_data_model/models/mapping/_classic2core.py,sha256=F0zusTh9pPR4z-RExPw3o4EMBSU2si6FJLuej2a3JzM,1430
 cognite/neat/v0/core/_data_model/models/mapping/_classic2core.yaml,sha256=ei-nuivNWVW9HmvzDBKIPF6ZdgaMq64XHw_rKm0CMxg,22584
 cognite/neat/v0/core/_data_model/models/physical/__init__.py,sha256=pH5ZF8jiW0A2w7VCSoHUsXxe894QFvTtgjxXNGVVaxk,990
-cognite/neat/v0/core/_data_model/models/physical/_exporter.py,sha256=ega1Yyosq-D9XYCypx9wtuMTFwS0EMRo_e3JwW434XU,30335
+cognite/neat/v0/core/_data_model/models/physical/_exporter.py,sha256=slECWjHinQscoR86dDmvB4OalyCvE9D_bSKp4v_GINg,30555
 cognite/neat/v0/core/_data_model/models/physical/_unverified.py,sha256=eE2D2DPfqEfOnYaUmOTPhBpHp9oN-gxjn8CTe_GKtz0,22419
 cognite/neat/v0/core/_data_model/models/physical/_validation.py,sha256=g0uSZImYaGmMZEwHUpb8KF9LTPiztz7k7cFQT3jrs4Y,41326
 cognite/neat/v0/core/_data_model/models/physical/_verified.py,sha256=49dn3K_gWO3VgfS-SK4kHpbSJywtj4mwES3iy-43z4M,29824
@@ -312,7 +313,7 @@ cognite/neat/v0/session/engine/__init__.py,sha256=D3MxUorEs6-NtgoICqtZ8PISQrjrr4
 cognite/neat/v0/session/engine/_import.py,sha256=1QxA2_EK613lXYAHKQbZyw2yjo5P9XuiX4Z6_6-WMNQ,169
 cognite/neat/v0/session/engine/_interface.py,sha256=3W-cYr493c_mW3P5O6MKN1xEQg3cA7NHR_ev3zdF9Vk,533
 cognite/neat/v0/session/engine/_load.py,sha256=u0x7vuQCRoNcPt25KJBJRn8sJabonYK4vtSZpiTdP4k,5201
-cognite_neat-0.127.27.dist-info/METADATA,sha256=QOiSm6GFoghApq_RDl6xHl3ACh9pnoLdrVRV_j-uAGk,9150
-cognite_neat-0.127.27.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-cognite_neat-0.127.27.dist-info/licenses/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
-cognite_neat-0.127.27.dist-info/RECORD,,
+cognite_neat-0.127.29.dist-info/METADATA,sha256=BODhl8sBHt2YtVcTIr7NK7Pcew1A68onQGB63HaKYDo,9150
+cognite_neat-0.127.29.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+cognite_neat-0.127.29.dist-info/licenses/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
+cognite_neat-0.127.29.dist-info/RECORD,,