cognite-neat 0.84.1__tar.gz → 0.85.0__tar.gz

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-neat
-Version: 0.84.1
+Version: 0.85.0
 Summary: Knowledge graph transformation
 Home-page: https://cognite-neat.readthedocs-hosted.com/
 License: Apache-2.0
@@ -16,6 +16,7 @@ Provides-Extra: all
 Provides-Extra: docs
 Provides-Extra: google
 Provides-Extra: graphql
+Provides-Extra: jupyter
 Provides-Extra: oxi
 Provides-Extra: service
 Requires-Dist: PyYAML
@@ -46,6 +47,7 @@ Requires-Dist: pymdown-extensions ; extra == "docs"
 Requires-Dist: pyoxigraph (==0.3.19) ; extra == "oxi" or extra == "all"
 Requires-Dist: rdflib
 Requires-Dist: requests
+Requires-Dist: rich[jupyter] (>=13.7.1,<14.0.0) ; extra == "jupyter"
 Requires-Dist: schedule (>=1,<2) ; extra == "service" or extra == "all"
 Requires-Dist: tomli (>=2.0.1,<3.0.0) ; python_version < "3.11"
 Requires-Dist: typing_extensions (>=4.8,<5.0) ; python_version < "3.11"

cognite_neat-0.85.0/cognite/neat/__init__.py

@@ -0,0 +1,4 @@
+from ._version import __version__
+from .utils.auth import get_cognite_client
+
+__all__ = ["__version__", "get_cognite_client"]

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/_shared.py

@@ -17,8 +17,24 @@ class NeatObject:
         """Return a dictionary representation of the object."""
         raise NotImplementedError()
 
+    def _repr_html_(self) -> str:
+        return pd.Series(self.dump(aggregate=True)).to_frame(name="value")._repr_html_()
+
+
+@dataclass(frozen=True)
+class FrozenNeatObject:
+    """A frozen neat object can be dumped to a dictionary."""
+
+    @abstractmethod
+    def dump(self, aggregate: bool = True) -> dict[str, Any]:
+        """Return a dictionary representation of the object."""
+        raise NotImplementedError()
+
+    def _repr_html_(self) -> str:
+        return pd.Series(self.dump(aggregate=True)).to_frame(name="value")._repr_html_()
+
 
-T_NeatObject = TypeVar("T_NeatObject", bound=NeatObject)
+T_NeatObject = TypeVar("T_NeatObject", bound=NeatObject | FrozenNeatObject)
 
 
 class NeatList(list, Sequence[T_NeatObject]):

cognite_neat-0.85.0/cognite/neat/_version.py

@@ -0,0 +1 @@
+__version__ = "0.85.0"

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/__init__.py

@@ -1,3 +1,4 @@
+from ._base import BaseExtractor
 from ._classic_cdf._assets import AssetsExtractor
 from ._classic_cdf._events import EventsExtractor
 from ._classic_cdf._files import FilesExtractor
@@ -10,6 +11,7 @@ from ._mock_graph_generator import MockGraphGenerator
 from ._rdf_file import RdfFileExtractor
 
 __all__ = [
+    "BaseExtractor",
     "AssetsExtractor",
     "MockGraphGenerator",
     "RelationshipsExtractor",
@@ -35,3 +37,23 @@ TripleExtractors = (
     | RdfFileExtractor
     | DexpiExtractor
 )
+
+
+def _repr_html_() -> str:
+    import pandas as pd
+
+    table = pd.DataFrame(  # type: ignore[operator]
+        [
+            {
+                "Extractor": name,
+                "Description": globals()[name].__doc__.strip().split("\n")[0] if globals()[name].__doc__ else "Missing",
+            }
+            for name in __all__
+            if name != "BaseExtractor"
+        ]
+    )._repr_html_()
+
+    return (
+        "<strong>Extractor</strong> An extractor is used to read data from "
+        f"a source into Neat's internal triple storage. <br />{table}"
+    )

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_base.py

@@ -2,6 +2,7 @@ from abc import abstractmethod
 from collections.abc import Iterable
 
 from cognite.neat.graph.models import Triple
+from cognite.neat.utils.auxiliary import class_html_doc
 
 
 class BaseExtractor:
@@ -12,3 +13,7 @@ class BaseExtractor:
     @abstractmethod
     def extract(self) -> Iterable[Triple]:
         raise NotImplementedError()
+
+    @classmethod
+    def _repr_html_(cls) -> str:
+        return class_html_doc(cls)

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_assets.py

@@ -14,6 +14,13 @@ from cognite.neat.utils.utils import create_sha256_hash, string_to_ideal_type
 
 
 class AssetsExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions Assets into Neat.
+
+    Args:
+        assets (Iterable[Asset]): An iterable of assets.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         assets: Iterable[Asset],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_events.py

@@ -15,6 +15,13 @@ from cognite.neat.utils.utils import string_to_ideal_type
 
 
 class EventsExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions Events into Neat.
+
+    Args:
+        events (Iterable[Event]): An iterable of events.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         events: Iterable[Event],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_labels.py

@@ -14,6 +14,13 @@ from cognite.neat.utils.utils import create_sha256_hash
 
 
 class LabelsExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions Labels into Neat.
+
+    Args:
+        labels (Iterable[LabelDefinition]): An iterable of labels.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         labels: Iterable[LabelDefinition],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_relationships.py

@@ -15,6 +15,13 @@ from cognite.neat.utils.utils import create_sha256_hash
 
 
 class RelationshipsExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions Relationships into Neat.
+
+    Args:
+        relationships (Iterable[Asset]): An iterable of relationships.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         relationships: Iterable[Relationship],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_sequences.py

@@ -15,6 +15,13 @@ from cognite.neat.utils.utils import string_to_ideal_type
 
 
 class SequencesExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions Sequences into Neat.
+
+    Args:
+        sequence (Iterable[Sequence]): An iterable of sequences.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         sequence: Iterable[Sequence],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_classic_cdf/_timeseries.py

@@ -15,6 +15,13 @@ from cognite.neat.utils.utils import string_to_ideal_type
 
 
 class TimeSeriesExtractor(BaseExtractor):
+    """Extract data from Cognite Data Fusions TimeSeries into Neat.
+
+    Args:
+        timeseries (Iterable[TimeSeries]): An iterable of timeseries.
+        namespace (Namespace, optional): The namespace to use. Defaults to DEFAULT_NAMESPACE.
+    """
+
     def __init__(
         self,
         timeseries: Iterable[TimeSeries],

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_dexpi.py

@@ -10,7 +10,7 @@ from cognite.neat.constants import DEFAULT_NAMESPACE
 from cognite.neat.graph.extractors._base import BaseExtractor
 from cognite.neat.graph.models import Triple
 from cognite.neat.utils.utils import as_neat_compliant_uri
-from cognite.neat.utils.xml import get_children, iterate_tree
+from cognite.neat.utils.xml_ import get_children, iterate_tree
 
 DEXPI = Namespace("http://sandbox.dexpi.org/rdl/")
 QUDT = Namespace("https://qudt.org/vocab/unit/")

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/extractors/_rdf_file.py

@@ -7,6 +7,14 @@ from cognite.neat.graph.extractors._base import BaseExtractor
 
 
 class RdfFileExtractor(BaseExtractor):
+    """Extract data from RDF files into Neat.
+
+    Args:
+        filepath (Path): The path to the RDF file.
+        mime_type (MIMETypes, optional): The MIME type of the RDF file. Defaults to "application/rdf+xml".
+        base_uri (URIRef, optional): The base URI to use. Defaults to None.
+    """
+
     def __init__(
         self,
         filepath: Path,

cognite_neat-0.85.0/cognite/neat/graph/loaders/__init__.py

@@ -0,0 +1,23 @@
+from ._base import BaseLoader, CDFLoader
+from ._rdf2dms import DMSLoader
+
+__all__ = ["BaseLoader", "CDFLoader", "DMSLoader"]
+
+
+def _repr_html_() -> str:
+    import pandas as pd
+
+    table = pd.DataFrame(  # type: ignore[operator]
+        [
+            {
+                "Loader": name,
+                "Description": globals()[name].__doc__.strip().split("\n")[0] if globals()[name].__doc__ else "Missing",
+            }
+            for name in __all__
+            if name not in ("BaseLoader", "CDFLoader")
+        ]
+    )._repr_html_()
+
+    return (
+        "<strong>Loader</strong> A loader writes data from Neat's triple storage into a target system" f"<br />{table}"
+    )

cognite_neat-0.85.0/cognite/neat/graph/loaders/_base.py

@@ -0,0 +1,86 @@
+from abc import ABC, abstractmethod
+from collections.abc import Hashable, Iterable
+from pathlib import Path
+from typing import ClassVar, Generic, TypeVar
+
+from cognite.client import CogniteClient
+from cognite.client.data_classes.capabilities import Capability
+
+from cognite.neat.graph import NeatGraphStore
+from cognite.neat.graph.issues.loader import FailedAuthorizationError
+from cognite.neat.issues import NeatIssue, NeatIssueList
+from cognite.neat.utils.auxiliary import class_html_doc
+from cognite.neat.utils.upload import UploadResult, UploadResultList
+
+T_Output = TypeVar("T_Output")
+
+
+class BaseLoader(ABC, Generic[T_Output]):
+    _new_line = "\n"
+    _encoding = "utf-8"
+
+    def __init__(self, graph_store: NeatGraphStore):
+        self.graph_store = graph_store
+
+    @abstractmethod
+    def write_to_file(self, filepath: Path) -> None:
+        raise NotImplementedError
+
+    def load(self, stop_on_exception: bool = False) -> Iterable[T_Output | NeatIssue]:
+        """Load the graph with data."""
+        return self._load(stop_on_exception)
+
+    @abstractmethod
+    def _load(self, stop_on_exception: bool = False) -> Iterable[T_Output | NeatIssue]:
+        """Load the graph with data."""
+        pass
+
+    @classmethod
+    def _repr_html_(cls) -> str:
+        return class_html_doc(cls)
+
+
+class CDFLoader(BaseLoader[T_Output]):
+    _UPLOAD_BATCH_SIZE: ClassVar[int] = 1000
+
+    def load_into_cdf(self, client: CogniteClient, dry_run: bool = False) -> UploadResultList:
+        return UploadResultList(self.load_into_cdf_iterable(client, dry_run))
+
+    def load_into_cdf_iterable(self, client: CogniteClient, dry_run: bool = False) -> Iterable[UploadResult]:
+        missing_capabilities = client.iam.verify_capabilities(self._get_required_capabilities())
+        if missing_capabilities:
+            upload_result = UploadResult[Hashable](name=type(self).__name__)
+            upload_result.issues.append(
+                FailedAuthorizationError(action="Upload to CDF", reason=str(missing_capabilities))
+            )
+            yield upload_result
+            return
+
+        issues = NeatIssueList[NeatIssue]()
+        items: list[T_Output] = []
+        for result in self.load(stop_on_exception=False):
+            if isinstance(result, NeatIssue):
+                issues.append(result)
+            else:
+                items.append(result)
+
+            if len(items) >= self._UPLOAD_BATCH_SIZE:
+                yield self._upload_to_cdf(client, items, dry_run, issues)
+                issues = NeatIssueList[NeatIssue]()
+                items = []
+        if items:
+            yield self._upload_to_cdf(client, items, dry_run, issues)
+
+    @abstractmethod
+    def _get_required_capabilities(self) -> list[Capability]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def _upload_to_cdf(
+        self,
+        client: CogniteClient,
+        items: list[T_Output],
+        dry_run: bool,
+        read_issues: NeatIssueList,
+    ) -> UploadResult:
+        raise NotImplementedError

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/loaders/_rdf2dms.py

@@ -10,6 +10,7 @@ from cognite.client import CogniteClient
 from cognite.client import data_modeling as dm
 from cognite.client.data_classes.capabilities import Capability, DataModelInstancesAcl
 from cognite.client.data_classes.data_modeling import ViewId
+from cognite.client.data_classes.data_modeling.ids import InstanceId
 from cognite.client.data_classes.data_modeling.views import SingleEdgeConnection
 from cognite.client.exceptions import CogniteAPIError
 from pydantic import ValidationInfo, create_model, field_validator
@@ -21,13 +22,24 @@ from cognite.neat.graph.stores import NeatGraphStore
 from cognite.neat.issues import NeatIssue, NeatIssueList
 from cognite.neat.rules.models import DMSRules
 from cognite.neat.rules.models.data_types import _DATA_TYPE_BY_DMS_TYPE
-from cognite.neat.utils.upload import UploadDiffsID
+from cognite.neat.utils.upload import UploadResult
 from cognite.neat.utils.utils import create_sha256_hash
 
 from ._base import CDFLoader
 
 
 class DMSLoader(CDFLoader[dm.InstanceApply]):
+    """Load data from Cognite Data Fusions Data Modeling Service (DMS) into Neat.
+
+    Args:
+        graph_store (NeatGraphStore): The graph store to load the data into.
+        data_model (dm.DataModel[dm.View] | None): The data model to load.
+        instance_space (str): The instance space to load the data into.
+        class_by_view_id (dict[ViewId, str] | None): A mapping from view id to class name. Defaults to None.
+        creat_issues (Sequence[NeatIssue] | None): A list of issues that occurred during reading. Defaults to None.
+        tracker (type[Tracker] | None): The tracker to use. Defaults to None.
+    """
+
     def __init__(
         self,
         graph_store: NeatGraphStore,
@@ -243,11 +255,10 @@ class DMSLoader(CDFLoader[dm.InstanceApply]):
         self,
         client: CogniteClient,
         items: list[dm.InstanceApply],
-        return_diffs: bool,
         dry_run: bool,
         read_issues: NeatIssueList,
-    ) -> UploadDiffsID:
-        result = UploadDiffsID(name=type(self).__name__, issues=read_issues)
+    ) -> UploadResult:
+        result = UploadResult[InstanceId](name=type(self).__name__, issues=read_issues)
         try:
             nodes = [item for item in items if isinstance(item, dm.NodeApply)]
             edges = [item for item in items if isinstance(item, dm.EdgeApply)]
@@ -260,16 +271,17 @@ class DMSLoader(CDFLoader[dm.InstanceApply]):
             )
         except CogniteAPIError as e:
             result.error_messages.append(str(e))
-            result.failed.append([repr(instance.as_id()) for instance in items])  # type: ignore[arg-type, attr-defined]
+            result.failed_upserted.update(item.as_id() for item in e.failed + e.unknown)
+            result.created.update(item.as_id() for item in e.successful)
         else:
             for instance in itertools.chain(upserted.nodes, upserted.edges):
                 if instance.was_modified and instance.created_time == instance.last_updated_time:
-                    result.created.append(repr(instance.as_id()))
+                    result.created.add(instance.as_id())
                 elif instance.was_modified:
-                    result.changed.append(repr(instance.as_id()))
+                    result.changed.add(instance.as_id())
                 else:
-                    result.unchanged.append(repr(instance.as_id()))
-        return result if return_diffs else result.as_upload_result_ids()  # type: ignore[return-value]
+                    result.unchanged.add(instance.as_id())
+        return result
 
 
 def _triples2dictionary(

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/stores/_base.py

@@ -282,3 +282,13 @@ class NeatGraphStore:
                     description=transformer.description,
                 )
             )
+
+    def _repr_html_(self) -> str:
+        provenance = self.provenance._repr_html_()
+
+        return (
+            f"<strong>{type(self).__name__}</strong> A graph store is a container for storing triples. "
+            "It can be queried and transformed to extract information.<br />"
+            "<strong>Provenance</strong> Provenance is a record of changes that have occurred in the graph store.<br />"
+            f"{provenance}"
+        )

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/graph/stores/_provenance.py

@@ -8,7 +8,6 @@
 
 
 import uuid
-from collections import UserList
 from collections.abc import Sequence
 from dataclasses import dataclass
 from datetime import datetime
@@ -16,6 +15,7 @@ from typing import TypeVar
 
 from rdflib import PROV, RDF, Literal, URIRef
 
+from cognite.neat._shared import FrozenNeatObject, NeatList
 from cognite.neat.constants import DEFAULT_NAMESPACE
 
 
@@ -64,7 +64,7 @@ class Entity:
 
 
 @dataclass(frozen=True)
-class Change:
+class Change(FrozenNeatObject):
     agent: Agent
     activity: Activity
     entity: Entity
@@ -81,11 +81,19 @@ class Change:
         entity = Entity(was_generated_by=activity, was_attributed_to=agent)
         return cls(agent, activity, entity, description)
 
+    def dump(self, aggregate: bool = True) -> dict[str, str]:
+        return {
+            "Agent": self.agent.id_,
+            "Activity": self.activity.id_,
+            "Entity": self.entity.id_,
+            "Description": self.description,
+        }
+
 
 T_Change = TypeVar("T_Change", bound=Change)
 
 
-class Provenance(UserList[T_Change]):
+class Provenance(NeatList[Change]):
     def __init__(self, changes: Sequence[T_Change] | None = None):
         super().__init__(changes or [])
 

cognite_neat-0.85.0/cognite/neat/rules/exporters/__init__.py

@@ -0,0 +1,37 @@
+from ._base import BaseExporter, CDFExporter
+from ._rules2dms import DMSExporter
+from ._rules2excel import ExcelExporter
+from ._rules2ontology import GraphExporter, OWLExporter, SemanticDataModelExporter, SHACLExporter
+from ._rules2yaml import YAMLExporter
+
+__all__ = [
+    "BaseExporter",
+    "DMSExporter",
+    "CDFExporter",
+    "SemanticDataModelExporter",
+    "OWLExporter",
+    "GraphExporter",
+    "SHACLExporter",
+    "ExcelExporter",
+    "YAMLExporter",
+]
+
+
+def _repr_html_() -> str:
+    import pandas as pd
+
+    table = pd.DataFrame(  # type: ignore[operator]
+        [
+            {
+                "Exporter": name,
+                "Description": globals()[name].__doc__.strip().split("\n")[0] if globals()[name].__doc__ else "Missing",
+            }
+            for name in __all__
+            if name not in ("BaseExporter", "CDFExporter", "GraphExporter")
+        ]
+    )._repr_html_()
+
+    return (
+        "<strong>Exporter</strong> An exporter converts Neat's representation of a data model called <em>Rules</em>"
+        f" into a schema/data model for a target format.<br />{table}"
+    )

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/exporters/_base.py

@@ -7,6 +7,7 @@ from cognite.client import CogniteClient
 
 from cognite.neat.rules._shared import Rules
 from cognite.neat.rules.models import DMSRules, InformationRules, RoleTypes
+from cognite.neat.utils.auxiliary import class_html_doc
 from cognite.neat.utils.upload import UploadResult, UploadResultList
 
 T_Export = TypeVar("T_Export")
@@ -34,6 +35,10 @@ class BaseExporter(ABC, Generic[T_Export]):
         else:
             raise NotImplementedError(f"Role {output_role} is not supported for {type(rules).__name__} rules")
 
+    @classmethod
+    def _repr_html_(cls) -> str:
+        return class_html_doc(cls, include_factory_methods=False)
+
 
 class CDFExporter(BaseExporter[T_Export]):
     @abstractmethod

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/exporters/_rules2dms.py

@@ -39,7 +39,7 @@ Component: TypeAlias = Literal["all", "spaces", "data_models", "views", "contain
 
 
 class DMSExporter(CDFExporter[DMSSchema]):
-    """Class for exporting rules object to CDF Data Model Storage (DMS).
+    """Export rules to Cognite Data Fusion's Data Model Storage (DMS) service.
 
     Args:
         export_components (frozenset[Literal["all", "spaces", "data_models", "views", "containers"]], optional):

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/exporters/_rules2ontology.py

@@ -39,16 +39,22 @@ class GraphExporter(BaseExporter[Graph], ABC):
 
 
 class OWLExporter(GraphExporter):
+    """Exports rules to an OWL ontology."""
+
     def export(self, rules: Rules) -> Graph:
         return Ontology.from_rules(rules).as_owl()
 
 
 class SHACLExporter(GraphExporter):
+    """Exports rules to a SHACL graph."""
+
     def export(self, rules: Rules) -> Graph:
         return Ontology.from_rules(rules).as_shacl()
 
 
 class SemanticDataModelExporter(GraphExporter):
+    """Exports rules to a semantic data model."""
+
     def export(self, rules: Rules) -> Graph:
         return Ontology.from_rules(rules).as_semantic_data_model()
 

cognite_neat-0.85.0/cognite/neat/rules/importers/__init__.py

@@ -0,0 +1,38 @@
+from ._base import BaseImporter
+from ._dms2rules import DMSImporter
+from ._dtdl2rules import DTDLImporter
+from ._inference2rules import InferenceImporter
+from ._owl2rules import OWLImporter
+from ._spreadsheet2rules import ExcelImporter, GoogleSheetImporter
+from ._yaml2rules import YAMLImporter
+
+__all__ = [
+    "BaseImporter",
+    "OWLImporter",
+    "DMSImporter",
+    "ExcelImporter",
+    "GoogleSheetImporter",
+    "DTDLImporter",
+    "YAMLImporter",
+    "InferenceImporter",
+]
+
+
+def _repr_html_() -> str:
+    import pandas as pd
+
+    table = pd.DataFrame(  # type: ignore[operator]
+        [
+            {
+                "Importer": name,
+                "Description": globals()[name].__doc__.strip().split("\n")[0] if globals()[name].__doc__ else "Missing",
+            }
+            for name in __all__
+            if name != "BaseImporter"
+        ]
+    )._repr_html_()
+
+    return (
+        "<strong>Importer</strong> An importer reads data/schema/data model from a source"
+        f" and converts it into Neat's representation of a data model called <em>Rules</em>.<br />{table}"
+    )

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/importers/_base.py

@@ -12,6 +12,7 @@ from rdflib import Namespace
 from cognite.neat.rules._shared import Rules
 from cognite.neat.rules.issues.base import IssueList, NeatValidationError, ValidationWarning
 from cognite.neat.rules.models import AssetRules, DMSRules, InformationRules, RoleTypes
+from cognite.neat.utils.auxiliary import class_html_doc
 
 
 class BaseImporter(ABC):
@@ -79,6 +80,10 @@ class BaseImporter(ABC):
             "description": f"Imported using {type(self).__name__}",
         }
 
+    @classmethod
+    def _repr_html_(cls) -> str:
+        return class_html_doc(cls)
+
 
 class _FutureResult:
     def __init__(self) -> None:

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/importers/_dms2rules.py

@@ -46,6 +46,16 @@ from cognite.neat.rules.models.entities import (
 
 
 class DMSImporter(BaseImporter):
+    """Imports a Data Model from Cognite Data Fusion.
+
+    Args:
+        schema: The schema containing the data model.
+        read_issues: A list of issues that occurred during the import.
+        metadata: Metadata for the data model.
+        ref_metadata: Metadata for the reference data model.
+
+    """
+
     def __init__(
         self,
         schema: DMSSchema,

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/importers/_dtdl2rules/dtdl_importer.py

@@ -18,7 +18,7 @@ from cognite.neat.utils.text import to_pascal
 
 
 class DTDLImporter(BaseImporter):
-    """Importer for DTDL (Digital Twin Definition Language).
+    """Importer from Azure Digital Twin - DTDL (Digital Twin Definition Language).
 
     This importer supports DTDL v2.0 and v3.0.
 

{cognite_neat-0.84.1 → cognite_neat-0.85.0}/cognite/neat/rules/importers/_inference2rules.py

@@ -33,7 +33,11 @@ INSTANCE_PROPERTIES_DEFINITION = """SELECT ?property (count(?property) as ?occur
 
 
 class InferenceImporter(BaseImporter):
-    """Rules inference through analysis of knowledge graph provided in various formats.
+    """Infers rules from a triple store.
+
+    Rules inference through analysis of knowledge graph provided in various formats.
+    Use the factory methods to create an triples store from sources such as
+    RDF files, JSON files, YAML files, XML files, or directly from a graph store.
 
     Args:
         issue_list: Issue list to store issues