cognite-neat 0.96.0__py3-none-any.whl → 0.96.2__py3-none-any.whl

cognite/neat/_constants.py

@@ -19,7 +19,17 @@ def _is_in_notebook() -> bool:
     return True
 
 
-IN_NOTEBOOK = _is_in_notebook()
+def _is_in_browser() -> bool:
+    try:
+        from pyodide.ffi import IN_BROWSER  # type: ignore [import-not-found]
+    except ModuleNotFoundError:
+        return False
+    return IN_BROWSER
+
+
+IN_PYODIDE = _is_in_browser()
+IN_NOTEBOOK = _is_in_notebook() or IN_PYODIDE
+
 
 PACKAGE_DIRECTORY = Path(neat.__file__).parent
 COGNITE_MODELS = (

cognite/neat/_rules/importers/_dms2rules.py

@@ -346,7 +346,7 @@ class DMSImporter(BaseImporter[DMSInputRules]):
             property_=prop_id,
             description=prop.description,
             name=prop.name,
-            connection=self._get_connection_type(prop_id, prop, view_entity.as_id()),
+            connection=self._get_connection_type(prop),
             value_type=str(value_type),
             is_list=self._get_is_list(prop),
             nullable=self._get_nullable(prop),
@@ -367,19 +367,13 @@ class DMSImporter(BaseImporter[DMSInputRules]):
         return self._all_containers_by_id[prop.container].properties[prop.container_property_identifier]
 
     def _get_connection_type(
-        self, prop_id: str, prop: ViewPropertyApply, view_id: dm.ViewId
+        self, prop: ViewPropertyApply
     ) -> Literal["direct"] | ReverseConnectionEntity | EdgeEntity | None:
-        if isinstance(prop, SingleEdgeConnectionApply | MultiEdgeConnectionApply) and prop.direction == "outwards":
+        if isinstance(prop, SingleEdgeConnectionApply | MultiEdgeConnectionApply):
             properties = ViewEntity.from_id(prop.edge_source) if prop.edge_source is not None else None
-            return EdgeEntity(properties=properties, type=DMSNodeEntity.from_reference(prop.type), direction="outwards")
-        elif isinstance(prop, SingleEdgeConnectionApply | MultiEdgeConnectionApply) and prop.direction == "inwards":
-            if reverse_prop := self._find_reverse_edge(prop_id, prop, view_id):
-                return ReverseConnectionEntity(property=reverse_prop)
-            else:
-                properties = ViewEntity.from_id(prop.source) if prop.edge_source is not None else None
-                return EdgeEntity(
-                    properties=properties, type=DMSNodeEntity.from_reference(prop.type), direction="inwards"
-                )
+            return EdgeEntity(
+                properties=properties, type=DMSNodeEntity.from_reference(prop.type), direction=prop.direction
+            )
         elif isinstance(prop, SingleReverseDirectRelationApply | MultiReverseDirectRelationApply):
             return ReverseConnectionEntity(property=prop.through.property)
         elif isinstance(prop, dm.MappedPropertyApply) and isinstance(

cognite/neat/_rules/models/dms/_exporter.py

@@ -565,7 +565,7 @@ class _DMSExporter:
             return edge_cls(
                 type=cls._create_edge_type_from_prop(prop),
                 source=source_view_id,
-                direction="outwards",
+                direction=prop.connection.direction,
                 name=prop.name,
                 description=prop.description,
                 edge_source=edge_source,
@@ -580,7 +580,6 @@ class _DMSExporter:
                     "If this error occurs it is a bug in NEAT, please report"
                     f"Debug Info, Invalid valueType reverse connection: {prop.model_dump_json()}"
                 )
-            edge_source = None
             reverse_prop = next(
                 (
                     prop
@@ -589,13 +588,6 @@ class _DMSExporter:
                 ),
                 None,
             )
-            if (
-                reverse_prop
-                and isinstance(reverse_prop.connection, EdgeEntity)
-                and reverse_prop.connection.properties is not None
-            ):
-                edge_source = reverse_prop.connection.properties.as_id()
-
             if reverse_prop is None:
                 warnings.warn(
                     PropertyNotFoundWarning(
@@ -608,30 +600,17 @@ class _DMSExporter:
                     stacklevel=2,
                 )
 
-            if reverse_prop is None or isinstance(reverse_prop.connection, EdgeEntity):
-                inwards_edge_cls = (
-                    dm.MultiEdgeConnectionApply if prop.is_list in [True, None] else SingleEdgeConnectionApply
-                )
-                return inwards_edge_cls(
-                    type=cls._create_edge_type_from_prop(reverse_prop or prop),
-                    source=source_view_id,
-                    name=prop.name,
-                    description=prop.description,
-                    direction="inwards",
-                    edge_source=edge_source,
-                )
-            elif reverse_prop and reverse_prop.connection == "direct":
-                reverse_direct_cls = (
-                    dm.MultiReverseDirectRelationApply
-                    if prop.is_list in [True, None]
-                    else SingleReverseDirectRelationApply
-                )
-                return reverse_direct_cls(
+            if reverse_prop and reverse_prop.connection == "direct":
+                args: dict[str, Any] = dict(
                     source=source_view_id,
                     through=dm.PropertyId(source=source_view_id, property=reverse_prop_id),
                     name=prop.name,
                     description=prop.description,
                 )
+                if prop.is_list in [True, None]:
+                    return dm.MultiReverseDirectRelationApply(**args)
+                else:
+                    return SingleReverseDirectRelationApply(**args)
             else:
                 return None
 

cognite/neat/_session/_base.py

@@ -1,12 +1,16 @@
 from typing import Literal, cast
 
 from cognite.client import CogniteClient
+from cognite.client import data_modeling as dm
 
 from cognite.neat import _version
 from cognite.neat._issues import IssueList
 from cognite.neat._rules import importers
 from cognite.neat._rules._shared import ReadRules
+from cognite.neat._rules.importers._rdf._base import DEFAULT_NON_EXISTING_NODE_TYPE
 from cognite.neat._rules.models import DMSRules
+from cognite.neat._rules.models.data_types import AnyURI
+from cognite.neat._rules.models.entities._single_value import UnknownEntity
 from cognite.neat._rules.models.information._rules import InformationRules
 from cognite.neat._rules.models.information._rules_input import InformationInputRules
 from cognite.neat._rules.transformers import ConvertToRules, VerifyAnyRules
@@ -51,7 +55,7 @@ class NeatSession:
                 self._state.store.add_rules(output.rules)
         self._state.issue_lists.append(output.issues)
         if output.issues:
-            print("You can inspect the issues with the .inspect attribute.")
+            print("You can inspect the issues with the .inspect.issues(...) method.")
         return output.issues
 
     def convert(self, target: Literal["dms"]) -> None:
@@ -62,15 +66,34 @@ class NeatSession:
 
     def infer(
         self,
-        space: str = "inference_space",
-        external_id: str = "InferredDataModel",
-        version: str = "v1",
+        model_id: dm.DataModelId | tuple[str, str, str] = (
+            "neat_space",
+            "NeatInferredDataModel",
+            "v1",
+        ),
+        non_existing_node_type: UnknownEntity | AnyURI = DEFAULT_NON_EXISTING_NODE_TYPE,
     ) -> IssueList:
-        input_rules: ReadRules = importers.InferenceImporter.from_graph_store(self._state.store).to_rules()
+        """Data model inference from instances.
 
-        cast(InformationInputRules, input_rules.rules).metadata.prefix = space
-        cast(InformationInputRules, input_rules.rules).metadata.name = external_id
-        cast(InformationInputRules, input_rules.rules).metadata.version = version
+        Args:
+            model_id: The ID of the inferred data model.
+            non_existing_node_type: The type of node to use when type of node is not possible to determine.
+        """
+
+        model_id = dm.DataModelId.load(model_id)
+
+        input_rules: ReadRules = importers.InferenceImporter.from_graph_store(
+            store=self._state.store,
+            non_existing_node_type=non_existing_node_type,
+        ).to_rules()
+
+        if model_id.space:
+            cast(InformationInputRules, input_rules.rules).metadata.prefix = model_id.space
+        if model_id.external_id:
+            cast(InformationInputRules, input_rules.rules).metadata.name = model_id.external_id
+
+        if model_id.version:
+            cast(InformationInputRules, input_rules.rules).metadata.version = model_id.version
 
         self.read.rdf._store_rules(self._state.store, input_rules, "Data Model Inference")
         return input_rules.issues

cognite/neat/_session/_show.py

@@ -3,9 +3,10 @@ import random
 from typing import Any, cast
 
 import networkx as nx
+from IPython.display import HTML, display
 from pyvis.network import Network as PyVisNetwork  # type: ignore
 
-from cognite.neat._constants import IN_NOTEBOOK
+from cognite.neat._constants import IN_NOTEBOOK, IN_PYODIDE
 from cognite.neat._rules._constants import EntityTypes
 from cognite.neat._rules.models.dms._rules import DMSRules
 from cognite.neat._rules.models.entities._single_value import ClassEntity, ViewEntity
@@ -31,6 +32,9 @@ class ShowBaseAPI:
         self._state = state
 
     def _generate_visualization(self, di_graph: nx.DiGraph, name: str) -> Any:
+        if not IN_NOTEBOOK:
+            raise NeatSessionError("Visualization is only available in Jupyter notebooks!")
+
         net = PyVisNetwork(
             notebook=IN_NOTEBOOK,
             cdn_resources="remote",
@@ -50,7 +54,12 @@ class ShowBaseAPI:
         )
 
         net.from_nx(di_graph)
-        return net.show(name)
+        if IN_PYODIDE:
+            net.write_html(name)
+            return display(HTML(name))
+
+        else:
+            return net.show(name)
 
 
 @intercept_session_exceptions

cognite/neat/_session/_to.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Any, overload
+from typing import Any, Literal, overload
 
 from cognite.client import CogniteClient
 
@@ -61,11 +61,17 @@ class CDFToAPI:
 
         return loader.load_into_cdf(self._client)
 
-    def data_model(self):
+    def data_model(self, existing_handling: Literal["fail", "skip", "update", "force"] = "skip"):
+        """Export the verified DMS data model to CDF.
+
+        Args:
+            existing_handling: How to handle if component of data model exists. Defaults to "skip".
+
+        """
         if not self._state.last_verified_dms_rules:
             raise ValueError("No verified DMS data model available")
 
-        exporter = exporters.DMSExporter()
+        exporter = exporters.DMSExporter(existing_handling=existing_handling)
 
         if not self._client:
             raise ValueError("No client provided!")

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.96.0"
+__version__ = "0.96.2"

{cognite_neat-0.96.0.dist-info → cognite_neat-0.96.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-neat
-Version: 0.96.0
+Version: 0.96.2
 Summary: Knowledge graph transformation
 Home-page: https://cognite-neat.readthedocs-hosted.com/
 License: Apache-2.0

{cognite_neat-0.96.0.dist-info → cognite_neat-0.96.2.dist-info}/RECORD

@@ -72,7 +72,7 @@ cognite/neat/_app/ui/neat-app/src/views/GlobalConfigView.tsx,sha256=1NMOby21Arrf
 cognite/neat/_app/ui/neat-app/src/views/WorkflowView.tsx,sha256=dU6xZip3MVaVzKAF23Cu6qZTl2Dn_evO-RTDAVGfONg,18451
 cognite/neat/_app/ui/neat-app/tsconfig.json,sha256=sw7AweQXRyJAIQ8Beft_380Q8zBr54pG1P_wutdPAIQ,664
 cognite/neat/_config.py,sha256=f9Py4SEHwYYquIg-k1rC7MbXBLENXQauoZtLyUbWvJQ,10118
-cognite/neat/_constants.py,sha256=rd61fuEEJcbyDIqSl5lh6QE7MBVqJiDmbfYUPptREPo,1649
+cognite/neat/_constants.py,sha256=_A-rsl18bPZCJmTSZ5jBvhO2k7kJrb2G9Hae1DEmuQ4,1888
 cognite/neat/_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognite/neat/_graph/_shared.py,sha256=WTDR46N3V8GZRZaqXcqR_IJnlMCQU7aT9xUmr_jBwFE,927
 cognite/neat/_graph/_tracking/__init__.py,sha256=pYj7c-YAUIP4hvN-4mlWnwaeZFerzL9_gM-oZhex7cE,91
@@ -152,7 +152,7 @@ cognite/neat/_rules/exporters/_rules2yaml.py,sha256=O9vnzDHf1ep1Qu0po0GVjgu945HN
 cognite/neat/_rules/exporters/_validation.py,sha256=DVJGdNNU2WtAFgUg0h4SWVhveRErEPOcYdT65y5toV0,682
 cognite/neat/_rules/importers/__init__.py,sha256=z682_ktGKDjr52DIL6cPvOercZS6-TYD_ZDo-MGqtck,1207
 cognite/neat/_rules/importers/_base.py,sha256=S-9Mi_58IvrIZ3oZaPRFnRWVIVhewcqGp8tng3q6kbw,2982
-cognite/neat/_rules/importers/_dms2rules.py,sha256=Hpjn8fPmH8f_Jnxx9RnS6nlqe-2JH1haiLy0ygg3nXU,22904
+cognite/neat/_rules/importers/_dms2rules.py,sha256=Q3L4N5b4-Db24EL-sttPBpVRWpvnRI-Tx_hSV86pTPo,22291
 cognite/neat/_rules/importers/_dtdl2rules/__init__.py,sha256=CNR-sUihs2mnR1bPMKs3j3L4ds3vFTsrl6YycExZTfU,68
 cognite/neat/_rules/importers/_dtdl2rules/_unit_lookup.py,sha256=wW4saKva61Q_i17guY0dc4OseJDQfqHy_QZBtm0OD6g,12134
 cognite/neat/_rules/importers/_dtdl2rules/dtdl_converter.py,sha256=BwClXc1ONzWacXbdoG5741Vj7knhJL1rRgyHh0X194Y,11911
@@ -185,7 +185,7 @@ cognite/neat/_rules/models/asset/_rules_input.py,sha256=-8s79O1DnwNQNPjJhcdzBlJv
 cognite/neat/_rules/models/asset/_validation.py,sha256=6u86qLlr1ehG1I5kIZhfCYTqUenuxUu9n_d7yWMUrS0,2021
 cognite/neat/_rules/models/data_types.py,sha256=ACQjS9vs7EjaNSO8R1kHiiNYbdeVQLCvxJYfZhWF8mM,9620
 cognite/neat/_rules/models/dms/__init__.py,sha256=CUqUlVjz4yZX_-61F-2ofSoV7N9MlSYx2N7vM-omp7E,640
-cognite/neat/_rules/models/dms/_exporter.py,sha256=dZAl5439p2K6LsA4yNpwCsEgNsxVkGjcmV9JGL3JAYE,30345
+cognite/neat/_rules/models/dms/_exporter.py,sha256=u9lLHwa6DUsh-xlaSbuU1YV5-xvbXObAocWvsRphGfc,29438
 cognite/neat/_rules/models/dms/_rules.py,sha256=JCYss9Nqf2tloyZj4gZktaW_45gMCjw7aLf5XeD_kRk,19980
 cognite/neat/_rules/models/dms/_rules_input.py,sha256=MjkkoEa5lRzQE0NUJ7TyQsrelWSoyGCCDFjMVG5pQvc,11530
 cognite/neat/_rules/models/dms/_schema.py,sha256=HSmSDvOm5S0x4Vb9tH9Jvd5i9tXiiM08E_Sdu6q_iA8,50783
@@ -212,14 +212,14 @@ cognite/neat/_rules/transformers/_mapping.py,sha256=RWHKPMaP3JdeCNvoDGu9ZGHxfyeI
 cognite/neat/_rules/transformers/_pipelines.py,sha256=-E5Hgitnr6ee8R9_3sqtjmWIPJ0w1xaLErG6Fo6ExVU,2603
 cognite/neat/_rules/transformers/_verification.py,sha256=330I1i3Va7Nau3DK3bTgEEGSHppmLvT-1yBMocDxV1A,4694
 cognite/neat/_session/__init__.py,sha256=fxQ5URVlUnmEGYyB8Baw7IDq-uYacqkigbc4b-Pr9Fw,58
-cognite/neat/_session/_base.py,sha256=3NB1yA7HsCIv5bN7b6IU6GsH0CDTzuHNjzokuL5jWKk,3691
+cognite/neat/_session/_base.py,sha256=b4hEcSde7n8_bW3DQ8sdRTo77O5GI5RWEuZ4QbkMOgA,4596
 cognite/neat/_session/_inspect.py,sha256=bucP5w2LKhmzijT222Nm6t0JOZ9K0qOFbK94mCJOTGo,2809
 cognite/neat/_session/_prepare.py,sha256=6mtJ6OkslSxWhe-xOIB569ml1RHeFUktuFWhrKzZ6Q8,4940
 cognite/neat/_session/_read.py,sha256=XVfns7ItV41znQQfEYLGi_U3eB2BtnIhkNXtcXte5b4,5563
 cognite/neat/_session/_set.py,sha256=DRvM5xAJLSs5DibdXFqs1h7yyepXSnHUwcPtfdsmx2w,874
-cognite/neat/_session/_show.py,sha256=Mwo1uFIeR58y3gmXPvZkbjg6wqBM08tMzpgJSPNykLw,8217
+cognite/neat/_session/_show.py,sha256=eeS0XRaPtY3GzELWcAQEvQkuMqO706Fcbqgvmt94lFg,8506
 cognite/neat/_session/_state.py,sha256=U3zPbKvTDAz7rd1yXdkWayY1p_5FJ6-0DIEw6fTD_vA,3292
-cognite/neat/_session/_to.py,sha256=3StR71QNEb8JKW-O2iI3aNTlT-wRRaabDg6dovMHWRw,2238
+cognite/neat/_session/_to.py,sha256=wz43PLhAR7FCntrhrIFr4e2jXhm9VKT7Qum3_IuELno,2536
 cognite/neat/_session/_wizard.py,sha256=Rdld2eZ-Q5BYbmAwW8FlfAYebdlw_P6L6V2WSDk-dHI,1306
 cognite/neat/_session/exceptions.py,sha256=tVearBcnz7gq0Wn5f7qssCUML_4oxrEvTCj6QnIcCxo,1265
 cognite/neat/_shared.py,sha256=RSaHm2eJceTlvb-hMMe4nHgoHdPYDfN3XcxDXo24k3A,1530
@@ -242,7 +242,7 @@ cognite/neat/_utils/text.py,sha256=PvTEsEjaTu8SE8yYaKUrce4msboMj933dK7-0Eey_rE,3
 cognite/neat/_utils/time_.py,sha256=O30LUiDH9TdOYz8_a9pFqTtJdg8vEjC3qHCk8xZblG8,345
 cognite/neat/_utils/upload.py,sha256=X5GGWHswnW0BrL2ulmm5MnKKtn-t1C8Ps3gb9Byc914,4016
 cognite/neat/_utils/xml_.py,sha256=FQkq84u35MUsnKcL6nTMJ9ajtG9D5i1u4VBnhGqP2DQ,1710
-cognite/neat/_version.py,sha256=B-kjJajURz8O11CWqC3d4KQ5aeL4lukk9XP51Xbpo8o,23
+cognite/neat/_version.py,sha256=h03CgwGcpDkIDSAlWJ_jyHT6Sh8R1n3OUcU2J2058f0,23
 cognite/neat/_workflows/__init__.py,sha256=S0fZq7kvoqDKodHu1UIPsqcpdvXoefUWRPt1lqeQkQs,420
 cognite/neat/_workflows/base.py,sha256=O1pcmfbme2gIVF2eOGrKZSUDmhZc8L9rI8UfvLN2YAM,26839
 cognite/neat/_workflows/cdf_store.py,sha256=3pebnATPo6In4-1srpa3wzstynTOi3T6hwFX5uaie4c,18050
@@ -271,8 +271,8 @@ cognite/neat/_workflows/tasks.py,sha256=dr2xuIb8P5e5e9p_fjzRlvDbKsre2xGYrkc3wnRx
 cognite/neat/_workflows/triggers.py,sha256=u69xOsaTtM8_WD6ZeIIBB-XKwvlZmPHAsZQh_TnyHcM,7073
 cognite/neat/_workflows/utils.py,sha256=gKdy3RLG7ctRhbCRwaDIWpL9Mi98zm56-d4jfHDqP1E,453
 cognite/neat/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognite_neat-0.96.0.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
-cognite_neat-0.96.0.dist-info/METADATA,sha256=uq_pivza7rnpyQzxjrD5VUDkY3gCC-RH_8XdfHQZZ_k,9573
-cognite_neat-0.96.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-cognite_neat-0.96.0.dist-info/entry_points.txt,sha256=SsQlnl8SNMSSjE3acBI835JYFtsIinLSbVmHmMEXv6E,51
-cognite_neat-0.96.0.dist-info/RECORD,,
+cognite_neat-0.96.2.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
+cognite_neat-0.96.2.dist-info/METADATA,sha256=BMCBJIE-W3M9MnQGoSxVZ0v3Wg2O8DcErGcN-0irhIQ,9573
+cognite_neat-0.96.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+cognite_neat-0.96.2.dist-info/entry_points.txt,sha256=SsQlnl8SNMSSjE3acBI835JYFtsIinLSbVmHmMEXv6E,51
+cognite_neat-0.96.2.dist-info/RECORD,,