cognite-neat 0.119.9__py3-none-any.whl → 0.121.0__py3-none-any.whl

cognite/neat/{_rules → core/_rules}/models/information/_rules_input.py

@@ -6,16 +6,16 @@ import pandas as pd
 from cognite.client import data_modeling as dm
 from rdflib import Namespace, URIRef
 
-from cognite.neat._constants import DEFAULT_NAMESPACE
-from cognite.neat._rules.models._base_input import InputComponent, InputRules
-from cognite.neat._rules.models.data_types import DataType
-from cognite.neat._rules.models.entities import (
+from cognite.neat.core._constants import DEFAULT_NAMESPACE
+from cognite.neat.core._rules.models._base_input import InputComponent, InputRules
+from cognite.neat.core._rules.models.data_types import DataType
+from cognite.neat.core._rules.models.entities import (
     ClassEntity,
     MultiValueTypeInfo,
     UnknownEntity,
     load_value_type,
 )
-from cognite.neat._utils.rdf_ import uri_display_name
+from cognite.neat.core._utils.rdf_ import uri_display_name
 
 from ._rules import (
     InformationClass,

cognite/neat/{_rules → core/_rules}/models/information/_validation.py

@@ -1,22 +1,22 @@
 import itertools
 from collections import Counter, defaultdict
 
-from cognite.neat._issues import IssueList
-from cognite.neat._issues.errors import NeatValueError
-from cognite.neat._issues.errors._resources import (
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._issues.errors import NeatValueError
+from cognite.neat.core._issues.errors._resources import (
     ResourceDuplicatedError,
     ResourceNotDefinedError,
 )
-from cognite.neat._issues.warnings._models import UndefinedClassWarning
-from cognite.neat._issues.warnings._resources import (
+from cognite.neat.core._issues.warnings._models import UndefinedClassWarning
+from cognite.neat.core._issues.warnings._resources import (
     ResourceNotDefinedWarning,
     ResourceRegexViolationWarning,
 )
-from cognite.neat._rules._constants import PATTERNS, EntityTypes
-from cognite.neat._rules.models.entities import ClassEntity, UnknownEntity
-from cognite.neat._rules.models.entities._multi_value import MultiValueTypeInfo
-from cognite.neat._utils.spreadsheet import SpreadsheetRead
-from cognite.neat._utils.text import humanize_collection
+from cognite.neat.core._rules._constants import PATTERNS, EntityTypes
+from cognite.neat.core._rules.models.entities import ClassEntity, UnknownEntity
+from cognite.neat.core._rules.models.entities._multi_value import MultiValueTypeInfo
+from cognite.neat.core._utils.spreadsheet import SpreadsheetRead
+from cognite.neat.core._utils.text import humanize_collection
 
 from ._rules import InformationRules
 

cognite/neat/{_rules → core/_rules}/models/mapping/_classic2core.py

@@ -3,9 +3,9 @@ from pathlib import Path
 
 import yaml
 
-from cognite.neat._issues.errors import NeatValueError
-from cognite.neat._rules._shared import ReadRules
-from cognite.neat._rules.models.dms import DMSInputRules, DMSRules
+from cognite.neat.core._issues.errors import NeatValueError
+from cognite.neat.core._rules._shared import ReadRules
+from cognite.neat.core._rules.models.dms import DMSInputRules, DMSRules
 
 _CLASSIC_TO_CORE_MAPPING = Path(__file__).resolve().parent / "_classic2core.yaml"
 
@@ -16,8 +16,8 @@ def _read_source_file() -> str:
 
 
 def load_classic_to_core_mapping(org_name: str | None, source_space: str, source_version: str) -> DMSRules:
-    from cognite.neat._rules.importers import YAMLImporter
-    from cognite.neat._rules.transformers import VerifyDMSRules
+    from cognite.neat.core._rules.importers import YAMLImporter
+    from cognite.neat.core._rules.transformers import VerifyDMSRules
 
     raw_str = _read_source_file()
     if org_name is not None:

cognite/neat/{_rules → core/_rules}/transformers/_base.py

@@ -4,10 +4,10 @@ from functools import lru_cache
 from types import UnionType
 from typing import Generic, TypeVar, Union, get_args, get_origin
 
-from cognite.neat._constants import DEFAULT_NAMESPACE
-from cognite.neat._rules._shared import ReadRules, Rules, VerifiedRules
-from cognite.neat._rules.models import DMSInputRules, InformationInputRules
-from cognite.neat._store._provenance import Agent as ProvenanceAgent
+from cognite.neat.core._constants import DEFAULT_NAMESPACE
+from cognite.neat.core._rules._shared import ReadRules, Rules, VerifiedRules
+from cognite.neat.core._rules.models import DMSInputRules, InformationInputRules
+from cognite.neat.core._store._provenance import Agent as ProvenanceAgent
 
 T_RulesIn = TypeVar("T_RulesIn", bound=Rules)
 T_RulesOut = TypeVar("T_RulesOut", bound=Rules)

cognite/neat/{_rules → core/_rules}/transformers/_converters.py

@@ -16,9 +16,12 @@ from cognite.client.utils.useful_types import SequenceNotStr
 from pydantic import ValidationError
 from rdflib import Namespace
 
-from cognite.neat._client import NeatClient
-from cognite.neat._client.data_classes.data_modeling import ContainerApplyDict, ViewApplyDict
-from cognite.neat._constants import (
+from cognite.neat.core._client import NeatClient
+from cognite.neat.core._client.data_classes.data_modeling import (
+    ContainerApplyDict,
+    ViewApplyDict,
+)
+from cognite.neat.core._constants import (
     COGNITE_CONCEPTS,
     COGNITE_MODELS,
     COGNITE_SPACES,
@@ -26,22 +29,25 @@ from cognite.neat._constants import (
     DMS_RESERVED_PROPERTIES,
     get_default_prefixes_and_namespaces,
 )
-from cognite.neat._issues import IssueList
-from cognite.neat._issues._factory import from_pydantic_errors
-from cognite.neat._issues.errors import CDFMissingClientError, NeatValueError
-from cognite.neat._issues.warnings import NeatValueWarning, PropertyOverwritingWarning
-from cognite.neat._issues.warnings._models import (
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._issues._factory import from_pydantic_errors
+from cognite.neat.core._issues.errors import CDFMissingClientError, NeatValueError
+from cognite.neat.core._issues.warnings import (
+    NeatValueWarning,
+    PropertyOverwritingWarning,
+)
+from cognite.neat.core._issues.warnings._models import (
     SolutionModelBuildOnTopOfCDMWarning,
 )
-from cognite.neat._rules._constants import PATTERNS, get_reserved_words
-from cognite.neat._rules._shared import (
+from cognite.neat.core._rules._constants import PATTERNS, get_reserved_words
+from cognite.neat.core._rules._shared import (
     ReadInputRules,
     ReadRules,
     VerifiedRules,
 )
-from cognite.neat._rules.analysis import RulesAnalysis
-from cognite.neat._rules.importers import DMSImporter
-from cognite.neat._rules.models import (
+from cognite.neat.core._rules.analysis import RulesAnalysis
+from cognite.neat.core._rules.importers import DMSImporter
+from cognite.neat.core._rules.models import (
     DMSInputRules,
     DMSRules,
     InformationInputRules,
@@ -49,10 +55,22 @@ from cognite.neat._rules.models import (
     SheetList,
     data_types,
 )
-from cognite.neat._rules.models.data_types import AnyURI, DataType, Enum, File, String, Timeseries
-from cognite.neat._rules.models.dms import DMSMetadata, DMSProperty, DMSValidation, DMSView
-from cognite.neat._rules.models.dms._rules import DMSContainer, DMSEnum, DMSNode
-from cognite.neat._rules.models.entities import (
+from cognite.neat.core._rules.models.data_types import (
+    AnyURI,
+    DataType,
+    Enum,
+    File,
+    String,
+    Timeseries,
+)
+from cognite.neat.core._rules.models.dms import (
+    DMSMetadata,
+    DMSProperty,
+    DMSValidation,
+    DMSView,
+)
+from cognite.neat.core._rules.models.dms._rules import DMSContainer, DMSEnum, DMSNode
+from cognite.neat.core._rules.models.entities import (
     ClassEntity,
     ContainerEntity,
     DMSUnknownEntity,
@@ -63,16 +81,22 @@ from cognite.neat._rules.models.entities import (
     UnknownEntity,
     ViewEntity,
 )
-from cognite.neat._rules.models.information import (
+from cognite.neat.core._rules.models.information import (
     InformationClass,
     InformationInputClass,
     InformationInputProperty,
     InformationMetadata,
     InformationProperty,
 )
-from cognite.neat._utils.rdf_ import get_inheritance_path
-from cognite.neat._utils.spreadsheet import SpreadsheetRead
-from cognite.neat._utils.text import NamingStandardization, humanize_collection, title, to_camel_case, to_words
+from cognite.neat.core._utils.rdf_ import get_inheritance_path
+from cognite.neat.core._utils.spreadsheet import SpreadsheetRead
+from cognite.neat.core._utils.text import (
+    NamingStandardization,
+    humanize_collection,
+    title,
+    to_camel_case,
+    to_words,
+)
 
 from ._base import RulesTransformer, T_VerifiedIn, T_VerifiedOut, VerifiedRulesTransformer
 from ._verification import VerifyDMSRules
@@ -1370,7 +1394,7 @@ class _InformationRulesConverter:
     def as_dms_rules(
         self, ignore_undefined_value_types: bool = False, reserved_properties: Literal["error", "warning"] = "error"
     ) -> "DMSRules":
-        from cognite.neat._rules.models.dms._rules import (
+        from cognite.neat.core._rules.models.dms._rules import (
             DMSContainer,
             DMSProperty,
             DMSRules,
@@ -1572,7 +1596,7 @@ class _InformationRulesConverter:
 
     @classmethod
     def _convert_metadata_to_dms(cls, metadata: InformationMetadata) -> "DMSMetadata":
-        from cognite.neat._rules.models.dms._rules import (
+        from cognite.neat.core._rules.models.dms._rules import (
             DMSMetadata,
         )
 
@@ -1598,7 +1622,7 @@ class _InformationRulesConverter:
         edge_value_types_by_class_property_pair: dict[tuple[ClassEntity, str], ClassEntity],
         end_node_by_edge: dict[ClassEntity, ClassEntity],
     ) -> "DMSProperty":
-        from cognite.neat._rules.models.dms._rules import DMSProperty
+        from cognite.neat.core._rules.models.dms._rules import DMSProperty
 
         # returns property type, which can be ObjectProperty or DatatypeProperty
         value_type = self._get_value_type(
@@ -1907,7 +1931,7 @@ class _DMSRulesConverter:
     def as_information_rules(
         self,
     ) -> "InformationRules":
-        from cognite.neat._rules.models.information._rules import (
+        from cognite.neat.core._rules.models.information._rules import (
             InformationClass,
             InformationProperty,
             InformationRules,
@@ -1987,7 +2011,9 @@ class _DMSRulesConverter:
 
     @classmethod
     def _convert_metadata_to_info(cls, metadata: DMSMetadata) -> "InformationMetadata":
-        from cognite.neat._rules.models.information._rules import InformationMetadata
+        from cognite.neat.core._rules.models.information._rules import (
+            InformationMetadata,
+        )
 
         return InformationMetadata(
             space=metadata.space,

cognite/neat/{_rules → core/_rules}/transformers/_mapping.py

@@ -5,13 +5,21 @@ from typing import Any, ClassVar, Literal
 
 from cognite.client import data_modeling as dm
 
-from cognite.neat._client import NeatClient
-from cognite.neat._issues.errors import CDFMissingClientError, NeatValueError, ResourceNotFoundError
-from cognite.neat._issues.warnings import PropertyOverwritingWarning
-from cognite.neat._rules.models import DMSRules, SheetList
-from cognite.neat._rules.models.data_types import Enum
-from cognite.neat._rules.models.dms import DMSContainer, DMSEnum, DMSProperty
-from cognite.neat._rules.models.entities import ClassEntity, ContainerEntity, ViewEntity
+from cognite.neat.core._client import NeatClient
+from cognite.neat.core._issues.errors import (
+    CDFMissingClientError,
+    NeatValueError,
+    ResourceNotFoundError,
+)
+from cognite.neat.core._issues.warnings import PropertyOverwritingWarning
+from cognite.neat.core._rules.models import DMSRules, SheetList
+from cognite.neat.core._rules.models.data_types import Enum
+from cognite.neat.core._rules.models.dms import DMSContainer, DMSEnum, DMSProperty
+from cognite.neat.core._rules.models.entities import (
+    ClassEntity,
+    ContainerEntity,
+    ViewEntity,
+)
 
 from ._base import VerifiedRulesTransformer
 

cognite/neat/{_rules → core/_rules}/transformers/_verification.py

@@ -1,22 +1,22 @@
 from abc import ABC
 
-from cognite.neat._client import NeatClient
-from cognite.neat._issues import MultiValueError, catch_issues
-from cognite.neat._issues.errors import NeatTypeError, NeatValueError
-from cognite.neat._rules._shared import (
+from cognite.neat.core._client import NeatClient
+from cognite.neat.core._issues import MultiValueError, catch_issues
+from cognite.neat.core._issues.errors import NeatTypeError, NeatValueError
+from cognite.neat.core._rules._shared import (
     ReadRules,
     T_ReadInputRules,
     T_VerifiedRules,
     VerifiedRules,
 )
-from cognite.neat._rules.models import (
+from cognite.neat.core._rules.models import (
     DMSInputRules,
     DMSRules,
     InformationInputRules,
     InformationRules,
 )
-from cognite.neat._rules.models.dms import DMSValidation
-from cognite.neat._rules.models.information import InformationValidation
+from cognite.neat.core._rules.models.dms import DMSValidation
+from cognite.neat.core._rules.models.information import InformationValidation
 
 from ._base import RulesTransformer
 

cognite/neat/{_store → core/_store}/_graph_store.py

@@ -12,16 +12,19 @@ from rdflib import Dataset, Graph, Namespace, URIRef
 from rdflib.graph import DATASET_DEFAULT_GRAPH_ID
 from rdflib.plugins.stores.sparqlstore import SPARQLUpdateStore
 
-from cognite.neat._graph._shared import quad_formats, rdflib_to_oxi_type
-from cognite.neat._graph.extractors import RdfFileExtractor, TripleExtractors
-from cognite.neat._graph.queries import Queries
-from cognite.neat._graph.transformers import Transformers
-from cognite.neat._issues import IssueList, catch_issues
-from cognite.neat._issues.errors import NeatValueError, OxigraphStorageLockedError
-from cognite.neat._shared import InstanceType, Triple
-from cognite.neat._utils.auxiliary import local_import
-from cognite.neat._utils.rdf_ import add_triples_in_batch, remove_namespace_from_uri
-from cognite.neat._utils.text import humanize_collection
+from cognite.neat.core._graph._shared import quad_formats, rdflib_to_oxi_type
+from cognite.neat.core._graph.extractors import RdfFileExtractor, TripleExtractors
+from cognite.neat.core._graph.queries import Queries
+from cognite.neat.core._graph.transformers import Transformers
+from cognite.neat.core._issues import IssueList, catch_issues
+from cognite.neat.core._issues.errors import NeatValueError, OxigraphStorageLockedError
+from cognite.neat.core._shared import InstanceType, Triple
+from cognite.neat.core._utils.auxiliary import local_import
+from cognite.neat.core._utils.rdf_ import (
+    add_triples_in_batch,
+    remove_namespace_from_uri,
+)
+from cognite.neat.core._utils.text import humanize_collection
 
 from ._provenance import Change, Entity, Provenance
 

cognite/neat/{_store → core/_store}/_provenance.py

@@ -24,9 +24,9 @@ from typing import Any, Generic, TypeVar
 
 from rdflib import PROV, RDF, Literal, URIRef
 
-from cognite.neat._constants import CDF_NAMESPACE, DEFAULT_NAMESPACE
-from cognite.neat._issues import IssueList
-from cognite.neat._shared import FrozenNeatObject, NeatList, Triple
+from cognite.neat.core._constants import CDF_NAMESPACE, DEFAULT_NAMESPACE
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._shared import FrozenNeatObject, NeatList, Triple
 
 
 @dataclass(frozen=True)

cognite/neat/{_store → core/_store}/_rules_store.py

@@ -10,18 +10,25 @@ from typing import Any, cast
 import rdflib
 from rdflib import URIRef
 
-from cognite.neat._client import NeatClient
-from cognite.neat._constants import DEFAULT_NAMESPACE
-from cognite.neat._graph.extractors import DMSGraphExtractor, KnowledgeGraphExtractor
-from cognite.neat._issues import IssueList, catch_issues
-from cognite.neat._issues.errors import NeatValueError
-from cognite.neat._rules._shared import T_VerifiedRules, VerifiedRules
-from cognite.neat._rules.exporters import BaseExporter
-from cognite.neat._rules.exporters._base import CDFExporter, T_Export
-from cognite.neat._rules.importers import BaseImporter
-from cognite.neat._rules.models import DMSRules, InformationRules
-from cognite.neat._rules.transformers import DMSToInformation, VerifiedRulesTransformer, VerifyAnyRules
-from cognite.neat._utils.upload import UploadResultList
+from cognite.neat.core._client import NeatClient
+from cognite.neat.core._constants import DEFAULT_NAMESPACE
+from cognite.neat.core._graph.extractors import (
+    DMSGraphExtractor,
+    KnowledgeGraphExtractor,
+)
+from cognite.neat.core._issues import IssueList, catch_issues
+from cognite.neat.core._issues.errors import NeatValueError
+from cognite.neat.core._rules._shared import T_VerifiedRules, VerifiedRules
+from cognite.neat.core._rules.exporters import BaseExporter
+from cognite.neat.core._rules.exporters._base import CDFExporter, T_Export
+from cognite.neat.core._rules.importers import BaseImporter
+from cognite.neat.core._rules.models import DMSRules, InformationRules
+from cognite.neat.core._rules.transformers import (
+    DMSToInformation,
+    VerifiedRulesTransformer,
+    VerifyAnyRules,
+)
+from cognite.neat.core._utils.upload import UploadResultList
 
 from ._provenance import (
     EXTERNAL_AGENT,

cognite/neat/{_store → core/_store}/exceptions.py

@@ -2,10 +2,10 @@
 
 from dataclasses import dataclass
 
-from cognite.neat._graph.extractors import KnowledgeGraphExtractor
-from cognite.neat._issues import IssueList
-from cognite.neat._rules.importers import BaseImporter
-from cognite.neat._rules.transformers import VerifiedRulesTransformer
+from cognite.neat.core._graph.extractors import KnowledgeGraphExtractor
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._rules.importers import BaseImporter
+from cognite.neat.core._rules.transformers import VerifiedRulesTransformer
 
 from ._provenance import Activity
 

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

@@ -11,7 +11,7 @@ from cognite.client.config import global_config
 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.core._utils.auxiliary import local_import
 
 __all__ = ["EnvironmentVariables", "get_cognite_client"]
 

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

@@ -11,7 +11,7 @@ from typing import Any
 
 from cognite.client.exceptions import CogniteDuplicatedError, CogniteReadTimeout
 
-from cognite.neat._issues.errors import NeatImportError
+from cognite.neat.core._issues.errors import NeatImportError
 
 
 def local_import(module: str, extra: str) -> ModuleType:

cognite/neat/{_utils → core/_utils}/collection_.py

@@ -2,8 +2,8 @@ from collections import Counter
 from collections.abc import Iterable, Sequence
 from typing import TypeVar
 
-from cognite.neat._config import GLOBAL_CONFIG
-from cognite.neat._constants import IN_PYODIDE
+from cognite.neat.core._config import GLOBAL_CONFIG
+from cognite.neat.core._constants import IN_PYODIDE
 
 T_Element = TypeVar("T_Element")
 

cognite/neat/{_utils → core/_utils}/graph_transformations_report.py

@@ -2,7 +2,7 @@ from abc import ABC
 from dataclasses import dataclass
 from typing import Any
 
-from cognite.neat._shared import NeatList, NeatObject
+from cognite.neat.core._shared import NeatList, NeatObject
 
 
 @dataclass

cognite/neat/{_utils → core/_utils}/rdf_.py

@@ -7,7 +7,7 @@ from pydantic import HttpUrl, TypeAdapter, ValidationError
 from rdflib import Graph, Namespace, URIRef
 from rdflib import Literal as RdfLiteral
 
-from cognite.neat._constants import SPACE_URI_PATTERN
+from cognite.neat.core._constants import SPACE_URI_PATTERN
 
 Triple: TypeAlias = tuple[URIRef, URIRef, RdfLiteral | URIRef]
 

cognite/neat/{_utils → core/_utils}/reader/_base.py

@@ -8,7 +8,7 @@ from urllib.parse import urlparse
 
 import requests
 
-from cognite.neat._issues.errors import NeatTypeError, NeatValueError
+from cognite.neat.core._issues.errors import NeatTypeError, NeatValueError
 
 
 class NeatReader(ABC):

cognite/neat/{_utils → core/_utils}/spreadsheet.py

@@ -8,7 +8,7 @@ from openpyxl import load_workbook
 from openpyxl.worksheet.datavalidation import DataValidation
 from openpyxl.worksheet.worksheet import Worksheet
 
-from cognite.neat._rules._constants import get_internal_properties
+from cognite.neat.core._rules._constants import get_internal_properties
 
 
 @dataclass
@@ -148,10 +148,17 @@ def find_column_with_value(sheet: Worksheet, value: Any) -> str | None:
     return None
 
 
-def generate_data_validation(sheet: str, column: str, no_header_rows: int, no_rows: int) -> DataValidation:
-    "Creates openpyxl data validation object for a cell in a sheet"
+def generate_data_validation(sheet: str, column: str, total_header_rows: int, validation_range: int) -> DataValidation:
+    """Creates openpyxl data validation object for a cell in a sheet
+
+    Args:
+        sheet: The name of the sheet where the data validation is applied.
+        column: The column letter where the data validation is applied.
+        total_header_rows: The number of header rows in the sheet.
+        validation_range: The total number of validation values in the column.
+    """
 
     return DataValidation(
         type="list",
-        formula1=f"={sheet}!{column}${no_header_rows + 1}:{column}${no_rows}",
+        formula1=f"={sheet}!{column}${total_header_rows + 1}:{column}${validation_range + total_header_rows + 1}",
     )

cognite/neat/{_utils → core/_utils}/text.py

@@ -4,7 +4,7 @@ from collections.abc import Collection, Set
 from re import Pattern
 from typing import Any
 
-from cognite.neat._rules._constants import get_reserved_words
+from cognite.neat.core._rules._constants import get_reserved_words
 
 PREPOSITIONS = frozenset(
     {

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

@@ -3,9 +3,9 @@ from dataclasses import dataclass, field
 from functools import total_ordering
 from typing import Any, Generic
 
-from cognite.neat._issues import IssueList
-from cognite.neat._issues.errors import NeatValueError
-from cognite.neat._shared import T_ID, NeatList, NeatObject
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._issues.errors import NeatValueError
+from cognite.neat.core._shared import T_ID, NeatList, NeatObject
 
 
 @total_ordering

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

@@ -5,22 +5,22 @@ from cognite.client import CogniteClient
 from cognite.client import data_modeling as dm
 
 from cognite.neat import _version
-from cognite.neat._client import NeatClient
-from cognite.neat._issues import IssueList
-from cognite.neat._issues.errors import RegexViolationError
-from cognite.neat._issues.errors._general import NeatImportError
-from cognite.neat._rules import importers
-from cognite.neat._rules.models import DMSRules
-from cognite.neat._rules.models.information._rules import InformationRules
-from cognite.neat._rules.transformers import (
+from cognite.neat.core._client import NeatClient
+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._rules import importers
+from cognite.neat.core._rules.models import DMSRules
+from cognite.neat.core._rules.models.information._rules import InformationRules
+from cognite.neat.core._rules.transformers import (
     InformationToDMS,
     MergeDMSRules,
     MergeInformationRules,
     ToDMSCompliantEntities,
     VerifyInformationRules,
 )
-from cognite.neat._store._rules_store import RulesEntity
-from cognite.neat._utils.auxiliary import local_import
+from cognite.neat.core._store._rules_store import RulesEntity
+from cognite.neat.core._utils.auxiliary import local_import
 
 from ._collector import _COLLECTOR, Collector
 from ._drop import DropAPI

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

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

cognite/neat/{_session → 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._constants import COGNITE_MODELS
-from cognite.neat._issues import IssueList
-from cognite.neat._rules.transformers import DropModelViews
+from cognite.neat.core._constants import COGNITE_MODELS
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._rules.transformers import DropModelViews
 
 from ._state import SessionState
 from .exceptions import NeatSessionError, session_class_wrapper

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

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

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

@@ -1,5 +1,5 @@
-from cognite.neat._issues._base import IssueList
-from cognite.neat._rules.transformers import (
+from cognite.neat.core._issues._base import IssueList
+from cognite.neat.core._rules.transformers import (
     ToCompliantEntities,
 )
 

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

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

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

@@ -1,6 +1,6 @@
-from cognite.neat._issues import IssueList
-from cognite.neat._rules.models.mapping import load_classic_to_core_mapping
-from cognite.neat._rules.transformers import (
+from cognite.neat.core._issues import IssueList
+from cognite.neat.core._rules.models.mapping import load_classic_to_core_mapping
+from cognite.neat.core._rules.transformers import (
     AsParentPropertyId,
     ChangeViewPrefix,
     IncludeReferenced,