cognite-neat 0.117.6__py3-none-any.whl → 0.117.8__py3-none-any.whl

cognite/neat/_rules/importers/_rdf/_base.py

@@ -1,8 +1,10 @@
 from datetime import datetime
 from pathlib import Path
+from typing import Any
 
 from cognite.client import data_modeling as dm
 from rdflib import Graph, Namespace, URIRef
+from typing_extensions import Self
 
 from cognite.neat._constants import get_default_prefixes_and_namespaces
 from cognite.neat._issues import IssueList, MultiValueError
@@ -69,7 +71,7 @@ class BaseRDFImporter(BaseImporter[InformationInputRules]):
         max_number_of_instance: int = -1,
         non_existing_node_type: UnknownEntity | AnyURI = DEFAULT_NON_EXISTING_NODE_TYPE,
         language: str = "en",
-    ):
+    ) -> Self:
         return cls(
             IssueList(title=f"{cls.__name__} issues"),
             store.dataset,
@@ -88,7 +90,7 @@ class BaseRDFImporter(BaseImporter[InformationInputRules]):
         non_existing_node_type: UnknownEntity | AnyURI = DEFAULT_NON_EXISTING_NODE_TYPE,
         language: str = "en",
         source_name: str = "Unknown",
-    ):
+    ) -> Self:
         issue_list = IssueList(title=f"{cls.__name__} issues")
 
         graph = Graph()
@@ -132,7 +134,7 @@ class BaseRDFImporter(BaseImporter[InformationInputRules]):
         raise NotImplementedError()
 
     @classmethod
-    def _add_uri_namespace_to_prefixes(cls, URI: URIRef, prefixes: dict[str, Namespace]):
+    def _add_uri_namespace_to_prefixes(cls: Any, URI: URIRef, prefixes: dict[str, Namespace]) -> None:
         """Add URI to prefixes dict if not already present
 
         Args:

cognite/neat/_rules/importers/_rdf/_inference2rules.py

@@ -264,24 +264,24 @@ class InferenceImporter(BaseRDFImporter):
                 property_["value_type"] = next(iter(property_["value_type"]))
 
         return {
-            "metadata": self._default_metadata().model_dump(),
+            "metadata": self._default_metadata(),
             "classes": list(classes.values()),
             "properties": list(properties.values()),
             "prefixes": prefixes,
         }
 
-    def _default_metadata(self):
+    def _default_metadata(self) -> dict[str, Any]:
         now = datetime.now(timezone.utc)
         return InformationMetadata(
             space=self.data_model_id.space,
             external_id=self.data_model_id.external_id,
-            version=self.data_model_id.version,
+            version=cast(str, self.data_model_id.version),
             name="Inferred Model",
-            creator="NEAT",
+            creator=["NEAT"],
             created=now,
             updated=now,
             description="Inferred model from knowledge graph",
-        )
+        ).model_dump()
 
     @property
     def source_uri(self) -> URIRef:

cognite/neat/_rules/importers/_rdf/_shared.py

@@ -151,7 +151,7 @@ def handle_meta(
     res: dict,
     feature: str,
     issue_list: IssueList,
-):
+) -> None:
     if not resources[resource_id][feature] and res[feature]:
         resources[resource_id][feature] = res[feature]
 

cognite/neat/_rules/models/_base_input.py

@@ -170,7 +170,7 @@ class InputComponent(ABC, Generic[T_RuleModel]):
                 args[field_name] = data[field_.alias]
         return cls(**args)
 
-    def dump(self, **kwargs) -> dict[str, Any]:
+    def dump(self, **kwargs: Any) -> dict[str, Any]:
         return {
             field_.alias or name: getattr(self, name)
             for name, field_ in self._get_verified_cls().model_fields.items()

cognite/neat/_rules/models/_base_rules.py

@@ -8,17 +8,7 @@ import types
 from abc import ABC, abstractmethod
 from collections.abc import Callable, Hashable, Iterator, MutableSequence, Sequence
 from datetime import datetime
-from typing import (
-    Annotated,
-    Any,
-    ClassVar,
-    Literal,
-    SupportsIndex,
-    TypeVar,
-    get_args,
-    get_origin,
-    overload,
-)
+from typing import Annotated, Any, ClassVar, Literal, SupportsIndex, TypeVar, get_args, get_origin, overload
 
 import pandas as pd
 from cognite.client import data_modeling as dm
@@ -128,7 +118,7 @@ class SchemaModel(BaseModel):
     )
 
     @classmethod
-    def mandatory_fields(cls, use_alias=False) -> set[str]:
+    def mandatory_fields(cls: Any, use_alias: bool = False) -> set[str]:
         """Returns a set of mandatory fields for the model."""
         return _get_required_fields(cls, use_alias)
 
@@ -193,7 +183,7 @@ class BaseMetadata(SchemaModel):
         return value
 
     @field_validator("description", mode="before")
-    def nan_as_none(cls, value):
+    def nan_as_none(cls: Any, value: Any) -> Any:
         if isinstance(value, float) and math.isnan(value):
             return None
         return value
@@ -207,7 +197,7 @@ class BaseMetadata(SchemaModel):
         return self.to_pandas().to_frame("value")._repr_html_()  # type: ignore[operator]
 
     @classmethod
-    def mandatory_fields(cls, use_alias=False) -> set[str]:
+    def mandatory_fields(cls: Any, use_alias: bool = False) -> set[str]:
         """Returns a set of mandatory fields for the model."""
         return _get_required_fields(cls, use_alias)
 
@@ -293,7 +283,7 @@ class BaseRules(SchemaModel, ABC):
         return headers_by_sheet
 
     @property
-    def display_name(self):
+    def display_name(self) -> str:
         return uri_display_name(self.metadata.identifier)
 
     def dump(

cognite/neat/_rules/models/_types.py

@@ -110,7 +110,7 @@ VersionType = Annotated[
 ]
 
 
-def _external_id_validation_factory(entity_type: EntityTypes, location: str):
+def _external_id_validation_factory(entity_type: EntityTypes, location: str) -> Callable:
     def _external_id_validation(value: str) -> str:
         compiled_regex = PATTERNS.entity_pattern(entity_type)
         if not compiled_regex.match(value):

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

@@ -253,7 +253,7 @@ class DMSInputView(InputComponent[DMSView]):
     neatId: str | URIRef | None = None
     logical: str | URIRef | None = None
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         if self.in_model is None:
             self.in_model = True
 
@@ -316,7 +316,7 @@ class DMSInputNode(InputComponent[DMSNode]):
     def from_node_type(cls, node_type: dm.NodeApply) -> "DMSInputNode":
         return cls(node=f"{node_type.space}:{node_type.external_id}", usage="type")
 
-    def dump(self, default_space: str, **_) -> dict[str, Any]:  # type: ignore[override]
+    def dump(self, default_space: str, **_) -> dict[str, Any]:  # type: ignore
         output = super().dump()
         output["Node"] = DMSNodeEntity.load(self.node, space=default_space)
         return output
@@ -366,7 +366,7 @@ class DMSInputRules(InputRules[DMSRules]):
         return "UnverifiedDMSModel"
 
     @property
-    def display_name(self):
+    def display_name(self) -> str:
         return uri_display_name(self.metadata.identifier)
 
     def _repr_html_(self) -> str:

cognite/neat/_rules/models/entities/_constants.py

@@ -8,7 +8,7 @@ class _UnknownType(BaseModel):
     def __str__(self) -> str:
         return "#N/A"
 
-    def __hash__(self):
+    def __hash__(self) -> int:
         return hash(str(self))
 
 

cognite/neat/_rules/models/entities/_multi_value.py

@@ -62,7 +62,7 @@ class MultiValueTypeInfo(BaseModel):
                 ]
             }
 
-    def set_default_prefix(self, prefix: str):
+    def set_default_prefix(self, prefix: str) -> None:
         for type_ in self.types:
             if isinstance(type_, ClassEntity) and type_.prefix is Undefined:
                 type_.prefix = prefix

cognite/neat/_rules/models/entities/_single_value.py

@@ -59,16 +59,16 @@ class Entity(BaseModel, extra="ignore"):
 
     @classmethod
     @overload
-    def load(cls: "type[T_Entity]", data: Any, strict: Literal[True], **defaults) -> "T_Entity": ...
+    def load(cls: "type[T_Entity]", data: Any, strict: Literal[True], **defaults: Any) -> "T_Entity": ...
 
     @classmethod
     @overload
     def load(
-        cls: "type[T_Entity]", data: Any, strict: Literal[False] = False, **defaults
+        cls: "type[T_Entity]", data: Any, strict: Literal[False] = False, **defaults: Any
     ) -> "T_Entity | UnknownEntity": ...
 
     @classmethod
-    def load(cls: "type[T_Entity]", data: Any, strict: bool = False, **defaults) -> "T_Entity | UnknownEntity":
+    def load(cls: "type[T_Entity]", data: Any, strict: bool = False, **defaults: Any) -> "T_Entity | UnknownEntity":
         if isinstance(data, cls):
             return data
         elif isinstance(data, str) and data == str(Unknown):
@@ -270,7 +270,7 @@ class ClassEntity(Entity):
     type_: ClassVar[EntityTypes] = EntityTypes.class_
     version: str | None = None
 
-    def as_view_entity(self, default_space: str, default_version) -> "ViewEntity":
+    def as_view_entity(self, default_space: str, default_version: str) -> "ViewEntity":
         if self.version is None:
             version = default_version
         else:
@@ -339,16 +339,18 @@ class DMSEntity(Entity, Generic[T_ID], ABC):
 
     @classmethod  # type: ignore[override]
     @overload
-    def load(cls: "type[T_DMSEntity]", data: Any, strict: Literal[True], **defaults) -> "T_DMSEntity": ...
+    def load(cls: "type[T_DMSEntity]", data: Any, strict: Literal[True], **defaults: Any) -> "T_DMSEntity": ...
 
     @classmethod
     @overload
     def load(
-        cls: "type[T_DMSEntity]", data: Any, strict: Literal[False] = False, **defaults
+        cls: "type[T_DMSEntity]", data: Any, strict: Literal[False] = False, **defaults: Any
     ) -> "T_DMSEntity | DMSUnknownEntity": ...
 
     @classmethod
-    def load(cls: "type[T_DMSEntity]", data: Any, strict: bool = False, **defaults) -> "T_DMSEntity | DMSUnknownEntity":  # type: ignore[override]
+    def load(
+        cls: "type[T_DMSEntity]", data: Any, strict: bool = False, **defaults: Any
+    ) -> "T_DMSEntity | DMSUnknownEntity":  # type: ignore
         if isinstance(data, str) and data == str(Unknown):
             if strict:
                 raise NeatValueError(f"Failed to load entity {data!s}")

cognite/neat/_rules/models/entities/_wrapped.py

@@ -1,5 +1,4 @@
 import json
-import re
 from abc import ABC, abstractmethod
 from collections.abc import Collection
 from functools import total_ordering
@@ -42,8 +41,8 @@ class WrappedEntity(BaseModel, ABC):
 
         # raw filter case:
         if cls.__name__ == "RawFilter":
-            if match := re.search(r"rawFilter\(([\s\S]*?)\)", data):
-                return {"filter": match.group(1), "inner": None}
+            if data.startswith("rawFilter(") and data.endswith(")"):
+                return {"filter": data[10:-1], "inner": None}
             else:
                 raise ValueError(f"Cannot parse {cls.name} from {data}. Ill formatted raw filter.")
 
@@ -57,7 +56,7 @@ class WrappedEntity(BaseModel, ABC):
     def as_str(self) -> str:
         return str(self)
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.id
 
     @property
@@ -193,7 +192,7 @@ class RawFilter(DMSFilter):
     filter: str
     inner: None = None  # type: ignore[assignment]
 
-    def as_dms_filter(self) -> dm.Filter:  # type: ignore[override]
+    def as_dms_filter(self, default: Any | None = None) -> dm.Filter:
         try:
             return dm.Filter.load(json.loads(self.filter))
         except json.JSONDecodeError as e:
@@ -206,5 +205,5 @@ class RawFilter(DMSFilter):
     def __repr__(self) -> str:
         return self.filter
 
-    def __str__(self):
+    def __str__(self) -> str:
         return f"{self.name}({self.filter})"

cognite/neat/_rules/models/information/_rules.py

@@ -186,26 +186,27 @@ class InformationProperty(SheetRow):
         return value
 
     @model_validator(mode="after")
-    def set_type_for_default(self):
+    def set_type_for_default(self) -> Any:
         if self.type_ == EntityTypes.data_property and self.default:
             default_value = self.default[0] if isinstance(self.default, list) else self.default
 
-            if type(default_value) is not self.value_type.python:
+            if type(default_value) is not self.value_type.python:  # type: ignore
                 try:
                     if isinstance(self.default, list):
                         updated_list = []
                         for value in self.default:
-                            updated_list.append(self.value_type.python(value))
+                            updated_list.append(self.value_type.python(value))  # type: ignore
                         self.default = updated_list
                     else:
-                        self.default = self.value_type.python(self.default)
+                        self.default = self.value_type.python(self.default)  # type: ignore
 
+                # this value_type.python does not seems correct. Need to check this further
                 except Exception:
                     raise PropertyDefinitionError(
                         self.class_,
                         "Class",
                         self.property_,
-                        f"Default value {self.default} is not of type {self.value_type.python}",
+                        f"Default value {self.default} is not of type {self.value_type.python}",  # type: ignore
                     ) from None
         return self
 

cognite/neat/_rules/models/information/_rules_input.py

@@ -43,7 +43,7 @@ class InformationInputMetadata(InputComponent[InformationMetadata]):
     def _get_verified_cls(cls) -> type[InformationMetadata]:
         return InformationMetadata
 
-    def dump(self, **kwargs) -> dict[str, Any]:
+    def dump(self, **kwargs: Any) -> dict[str, Any]:
         output = super().dump()
         if self.created is None:
             output["created"] = datetime.now()
@@ -97,7 +97,7 @@ class InformationInputProperty(InputComponent[InformationProperty]):
     def _get_verified_cls(cls) -> type[InformationProperty]:
         return InformationProperty
 
-    def dump(self, default_prefix: str, **kwargs) -> dict[str, Any]:  # type: ignore[override]
+    def dump(self, default_prefix: str, **kwargs) -> dict[str, Any]:  # type: ignore
         output = super().dump()
         output["Class"] = ClassEntity.load(self.class_, prefix=default_prefix)
         output["Value Type"] = load_value_type(self.value_type, default_prefix)
@@ -124,7 +124,7 @@ class InformationInputClass(InputComponent[InformationClass]):
     def class_str(self) -> str:
         return str(self.class_)
 
-    def dump(self, default_prefix: str, **kwargs) -> dict[str, Any]:  # type: ignore[override]
+    def dump(self, default_prefix: str, **kwargs) -> dict[str, Any]:  # type: ignore
         output = super().dump()
         parent: list[ClassEntity] | None = None
         if isinstance(self.implements, str):
@@ -163,7 +163,7 @@ class InformationInputRules(InputRules[InformationRules]):
         return "UnverifiedInformationModel"
 
     @property
-    def display_name(self):
+    def display_name(self) -> str:
         return uri_display_name(self.metadata.identifier)
 
     def _repr_html_(self) -> str:

cognite/neat/_rules/transformers/_mapping.py

@@ -346,7 +346,8 @@ class AsParentPropertyId(VerifiedRulesTransformer[DMSRules, DMSRules]):
 
     @staticmethod
     def _get_parent_view_property_by_container_property(
-        path_by_view, view_by_container_properties: dict[tuple[ContainerEntity, str], list[tuple[ViewEntity, str]]]
+        path_by_view: dict[ViewEntity, list[ViewEntity]],
+        view_by_container_properties: dict[tuple[ContainerEntity, str], list[tuple[ViewEntity, str]]],
     ) -> dict[tuple[ContainerEntity, str], str]:
         parent_name_by_container_property: dict[tuple[ContainerEntity, str], str] = {}
         for (container, container_property), view_properties in view_by_container_properties.items():

cognite/neat/_session/_collector.py

@@ -49,7 +49,7 @@ class Collector:
             return "notebook"
         return "python"
 
-    def track_session_command(self, command: str, *args, **kwargs) -> None:
+    def track_session_command(self, command: str, *args: Any, **kwargs: Any) -> None:
         event_information = {
             "neatVersion": __version__,
             "$os": platform.system(),

cognite/neat/_session/_read.py

@@ -616,7 +616,7 @@ class XMLReadAPI(BaseReadAPI):
         for transformer in transformers:
             self._state.instances.store.transform(cast(Transformers, transformer))
 
-    def aml(self, io: Any):
+    def aml(self, io: Any) -> None:
         """Reads an AML file into NeatSession and executes a set of predefined transformations.
 
         Args:

cognite/neat/_session/exceptions.py

@@ -30,9 +30,9 @@ class NeatSessionError(Exception):
     ...
 
 
-def _session_method_wrapper(func: Callable, cls_name: str):
+def _session_method_wrapper(func: Callable, cls_name: str) -> Any:
     @functools.wraps(func)
-    def wrapper(*args: Any, **kwargs: Any):
+    def wrapper(*args: Any, **kwargs: Any) -> Any:
         _COLLECTOR.track_session_command(f"{cls_name}.{func.__name__}", *args, **kwargs)
         try:
             with warnings.catch_warnings(record=True) as w:
@@ -59,7 +59,7 @@ def _session_method_wrapper(func: Callable, cls_name: str):
             else:
                 raise e
 
-    def _get_action():
+    def _get_action() -> str:
         action = func.__name__
         if action == "__call__":
             action = func.__qualname__.removesuffix(".__call__").removesuffix("API")
@@ -68,7 +68,7 @@ def _session_method_wrapper(func: Callable, cls_name: str):
     return wrapper
 
 
-def session_class_wrapper(cls: type):
+def session_class_wrapper(cls: type) -> type:
     """This decorator wraps all methods of a class.
 
     It should be used with all composition classes used with the NeatSession class.

cognite/neat/_store/_provenance.py

@@ -20,7 +20,7 @@ import uuid
 from collections.abc import Iterable, Sequence
 from dataclasses import dataclass, field
 from datetime import datetime
-from typing import Generic, TypeVar
+from typing import Any, Generic, TypeVar
 
 from rdflib import PROV, RDF, Literal, URIRef
 
@@ -178,10 +178,10 @@ class Provenance(NeatList[Change[T_Entity]]):
     def activity_took_place(self, activity: str) -> bool:
         return any(change.activity.used == activity for change in self)
 
-    def __delitem__(self, *args, **kwargs):
+    def __delitem__(self, *args: Any, **kwargs: Any) -> None:
         raise TypeError("Cannot delete change from provenance")
 
-    def __setitem__(self, *args, **kwargs):
+    def __setitem__(self, *args: Any, **kwargs: Any) -> None:
         raise TypeError("Cannot modify change from provenance")
 
     def _repr_html_(self) -> str:

cognite/neat/_store/_rules_store.py

@@ -221,7 +221,7 @@ class NeatRulesStore:
         for agent_tool in transformer:
 
             def action(
-                transformer_item=agent_tool,
+                transformer_item: VerifiedRulesTransformer = agent_tool,
             ) -> tuple[InformationRules, DMSRules | None]:
                 last_change = self.provenance[-1]
                 source_entity = last_change.target_entity
@@ -255,7 +255,7 @@ class NeatRulesStore:
         self,
         action: Callable[[], tuple[InformationRules, DMSRules | None]],
         agent_tool: BaseImporter | VerifiedRulesTransformer | KnowledgeGraphExtractor,
-    ):
+    ) -> IssueList:
         result, issue_list, start, end = self._do_activity(action)
         self._last_issues = issue_list
 
@@ -312,7 +312,7 @@ class NeatRulesStore:
     def _do_activity(
         self,
         action: Callable[[], tuple[InformationRules, DMSRules | None]],
-    ):
+    ) -> tuple[tuple[InformationRules, DMSRules | None], IssueList, datetime, datetime]:
         """This private method is used to execute an activity and return the result and issues."""
         start = datetime.now(timezone.utc)
         result: tuple[InformationRules, DMSRules | None] | None = None

cognite/neat/_store/exceptions.py

@@ -13,7 +13,7 @@ from ._provenance import Activity
 class NeatStoreError(Exception):
     """Base class for all exceptions in the store module"""
 
-    def __str__(self):
+    def __str__(self) -> str:
         return type(self).__name__
 
 
@@ -30,7 +30,7 @@ class ActivityFailed(NeatStoreError):
         self.issue_list = issue_list
         self.tool = tool
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.tool.description
 
 
@@ -49,7 +49,7 @@ class InvalidActivityOutput(NeatStoreError):
         self.activity = activity
         self.output = output
 
-    def __str__(self):
+    def __str__(self) -> str:
         return f"{super().__str__()}: {self.activity.id_} -> {self.output}"
 
 

cognite/neat/_utils/auth.py

@@ -106,7 +106,7 @@ class EnvironmentVariables:
     CDF_TIMEOUT: int | None = None
     CDF_REDIRECT_PORT: int = 53_000
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         if self.LOGIN_FLOW.lower() not in _VALID_LOGIN_FLOWS:
             raise ValueError(f"LOGIN_FLOW must be one of {_VALID_LOGIN_FLOWS}")
         if self.IDP_TOKEN_URL and not self.IDP_TENANT_ID:

cognite/neat/_utils/auxiliary.py

@@ -7,6 +7,7 @@ from collections.abc import Callable
 from datetime import datetime
 from functools import wraps
 from types import ModuleType
+from typing import Any
 
 from cognite.client.exceptions import CogniteDuplicatedError, CogniteReadTimeout
 
@@ -42,10 +43,10 @@ def class_html_doc(cls: type, include_factory_methods: bool = True) -> str:
     return f"<h3>{cls.__name__}</h3><p>{docstring}</p>"
 
 
-def retry_decorator(max_retries=2, retry_delay=3, component_name=""):
-    def decorator(func):
+def retry_decorator(max_retries: int = 2, retry_delay: int = 3, component_name: str = "") -> Any:
+    def decorator(func: Callable) -> Any:
         @wraps(func)
-        def wrapper(*args, **kwargs):
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
             previous_exception = None
             for attempt in range(max_retries + 1):
                 try:
@@ -89,7 +90,8 @@ def retry_decorator(max_retries=2, retry_delay=3, component_name=""):
                         raise e
 
                 except Exception as e:
-                    previous_exception = e
+                    # Deliberately ignoring here, this should be refactored.
+                    previous_exception = e  # type: ignore
                     if attempt < max_retries:
                         logging.error(
                             f"Retry attempt {attempt + 1} failed for {component_name}. "

cognite/neat/_utils/rdf_.py

@@ -67,7 +67,7 @@ def remove_namespace_from_uri(
         # Assume that all elements in the tuple are of the same type following type hint
         uris = URI
     else:
-        raise TypeError(f"URI must be of type URIRef or str, got {type(URI)}")
+        raise TypeError(f"URI must be of type URIRef or str, got {type(URI)}: {URI}")
 
     output = []
     for u in uris:
@@ -218,7 +218,7 @@ def add_triples_in_batch(graph: Graph, triples: Iterable[Triple], batch_size: in
     commit_counter = 0
     number_of_written_triples = 0
 
-    def check_commit(force_commit: bool = False):
+    def check_commit(force_commit: bool = False) -> None:
         """Commit nodes to the graph if batch counter is reached or if force_commit is True"""
         nonlocal commit_counter
         nonlocal number_of_written_triples
@@ -248,7 +248,7 @@ def remove_triples_in_batch(graph: Graph, triples: Iterable[Triple], batch_size:
     """
     batch_count = 0
 
-    def check_commit(force_commit: bool = False):
+    def check_commit(force_commit: bool = False) -> None:
         """Commit nodes to the graph if batch counter is reached or if force_commit is True"""
         nonlocal batch_count
         batch_count += 1
@@ -274,7 +274,7 @@ def remove_instance_ids_in_batch(graph: Graph, instance_ids: Iterable[URIRef], b
     """
     batch_count = 0
 
-    def check_commit(force_commit: bool = False):
+    def check_commit(force_commit: bool = False) -> None:
         """Commit nodes to the graph if batch counter is reached or if force_commit is True"""
         nonlocal batch_count
         batch_count += 1

cognite/neat/_utils/reader/_base.py

@@ -98,7 +98,7 @@ class PathReader(NeatReader):
         self._io = file
         return file
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(self) -> None:
         if self._io:
             self._io.close()
 

cognite/neat/_utils/spreadsheet.py

@@ -24,7 +24,7 @@ class SpreadsheetRead:
     skipped_rows: list[int] = field(default_factory=list)
     is_one_indexed: bool = True
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         self.empty_rows = sorted(self.empty_rows)
 
     def adjusted_row_number(self, row_no: int) -> int:

cognite/neat/_utils/text.py

@@ -63,7 +63,7 @@ def to_camel_case(string: str) -> str:
     return _to_camel_case(string, is_all_upper, is_first_upper)
 
 
-def _to_camel_case(string, is_all_upper: bool, is_first_upper: bool):
+def _to_camel_case(string: str, is_all_upper: bool, is_first_upper: bool) -> str:
     if "_" in string:
         pascal_splits = [
             _to_pascal_case(part, is_all_upper, is_first_upper and no == 0)

cognite/neat/_utils/time_.py

@@ -13,5 +13,5 @@ def epoch_now_ms() -> int:
     return int((datetime.now(UTC) - datetime(1970, 1, 1, tzinfo=UTC)).total_seconds() * 1000)
 
 
-def datetime_utc_now():
+def datetime_utc_now() -> datetime:
     return datetime.now(UTC)

cognite/neat/_version.py

@@ -1,2 +1,2 @@
-__version__ = "0.117.6"
+__version__ = "0.117.8"
 __engine__ = "^2.0.4"

{cognite_neat-0.117.6.dist-info → cognite_neat-0.117.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cognite-neat
-Version: 0.117.6
+Version: 0.117.8
 Summary: Knowledge graph transformation
 License: Apache-2.0
 Author: Nikola Vasiljevic