cognite-neat 0.102.0__py3-none-any.whl → 0.103.1__py3-none-any.whl

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

@@ -1,5 +1,6 @@
 import warnings
 from collections import Counter, defaultdict
+from collections.abc import Mapping
 from functools import lru_cache
 from typing import ClassVar
 
@@ -115,7 +116,9 @@ class DMSValidation:
             **ref_container_by_id,
         }
         all_views_by_id: dict[dm.ViewId, dm.ViewApply | dm.View] = {**dict(dms_schema.views.items()), **ref_view_by_id}
-        properties_by_ids = self._as_properties_by_ids(dms_schema, ref_view_by_id)
+        properties_by_ids = self._as_properties_by_ids(dms_schema.views, ref_view_by_id)
+        ref_properties_by_ids = self._as_properties_by_ids(ref_view_by_id, {})
+        all_properties_by_ids = {**ref_properties_by_ids, **properties_by_ids}
         view_properties_by_id = self._as_view_properties_by_id(properties_by_ids)
         parents_view_ids_by_child_id = self._parent_view_ids_by_child_id(all_views_by_id)
 
@@ -132,7 +135,9 @@ class DMSValidation:
         # SDK classes validation
         issue_list.extend(self._containers_are_proper_size(dms_schema))
         issue_list.extend(
-            self._validate_reverse_connections(properties_by_ids, all_containers_by_id, parents_view_ids_by_child_id)
+            self._validate_reverse_connections(
+                properties_by_ids, all_containers_by_id, parents_view_ids_by_child_id, all_properties_by_ids
+            )
         )
         issue_list.extend(self._validate_schema(dms_schema, all_views_by_id, all_containers_by_id))
         issue_list.extend(self._validate_referenced_container_limits(dms_schema.views, view_properties_by_id))
@@ -140,22 +145,22 @@ class DMSValidation:
 
     @staticmethod
     def _as_properties_by_ids(
-        dms_schema: DMSSchema, ref_view_by_id: dict[dm.ViewId, dm.View]
+        view_by_id: Mapping[dm.ViewId, dm.ViewApply] | Mapping[dm.ViewId, dm.View],
+        ref_view_by_id: dict[dm.ViewId, dm.View],
     ) -> dict[tuple[ViewId, str], ViewPropertyApply | ViewProperty]:
         # Priority DMS schema properties.
         # No need to do long lookups in ref_views as these already contain all ancestor properties.
         properties_by_id: dict[tuple[ViewId, str], ViewPropertyApply | ViewProperty] = {}
-        for view in dms_schema.views.values():
-            view_id = view.as_id()
+        for view_id, view in view_by_id.items():
             for prop_id, prop in (view.properties or {}).items():
                 properties_by_id[(view_id, prop_id)] = prop
             if view.implements:
                 to_check = view.implements.copy()
                 while to_check:
                     parent_id = to_check.pop()
-                    if parent_id in dms_schema.views:
-                        # Priority DMS Schema properties
-                        parent_view = dms_schema.views[parent_id]
+                    if parent_id in view_by_id:
+                        # Priority of the DMS schema properties
+                        parent_view = view_by_id[parent_id]
                         for prop_id, prop in (parent_view.properties or {}).items():
                             if (view_id, prop_id) not in properties_by_id:
                                 properties_by_id[(view_id, prop_id)] = prop
@@ -189,6 +194,8 @@ class DMSValidation:
     ) -> dict[ViewId, set[ViewId]]:
         @lru_cache
         def get_parents(child_view_id: ViewId) -> set[ViewId]:
+            if child_view_id not in all_views_by_id:
+                return set()
             child_view = all_views_by_id[child_view_id]
             parents = set(child_view.implements or [])
             for parent_id in child_view.implements or []:
@@ -406,6 +413,7 @@ class DMSValidation:
         view_property_by_property_id: dict[tuple[dm.ViewId, str], ViewPropertyApply | ViewProperty],
         containers_by_id: dict[dm.ContainerId, dm.ContainerApply | dm.Container],
         parents_by_view: dict[dm.ViewId, set[dm.ViewId]],
+        all_view_property_by_property_id: dict[tuple[dm.ViewId, str], ViewPropertyApply | ViewProperty],
     ) -> IssueList:
         issue_list = IssueList()
         # do not check for reverse connections in Cognite models
@@ -416,7 +424,7 @@ class DMSValidation:
             if not isinstance(prop_, ReverseDirectRelationApply | ReverseDirectRelation):
                 continue
             target_id = prop_.through.source, prop_.through.property
-            if target_id not in view_property_by_property_id:
+            if target_id not in all_view_property_by_property_id:
                 issue_list.append(
                     ReversedConnectionNotFeasibleError(
                         view_id,
@@ -430,7 +438,7 @@ class DMSValidation:
                 # Todo: How to handle this case? Should not happen if you created the model with Neat
                 continue
 
-            target_property = view_property_by_property_id[(target_id[0], target_id[1])]
+            target_property = all_view_property_by_property_id[(target_id[0], target_id[1])]
             # Validate that the target is a direct relation pointing to the view_id
             is_direct_relation = False
             if isinstance(target_property, dm.MappedProperty) and isinstance(target_property.type, dm.DirectRelation):

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

@@ -25,38 +25,38 @@ from ._types import CdfResourceEntityList, ClassEntityList, ContainerEntityList,
 from ._wrapped import DMSFilter, HasDataFilter, NodeTypeFilter, RawFilter, WrappedEntity
 
 __all__ = [
-    "Entity",
-    "UnitEntity",
-    "DMSVersionedEntity",
-    "RelationshipEntity",
-    "EdgeEntity",
-    "ViewEntity",
-    "ClassEntity",
-    "ViewEntityList",
-    "UnknownEntity",
-    "Unknown",
-    "Undefined",
-    "T_Entity",
     "AssetEntity",
     "AssetFields",
-    "ReverseConnectionEntity",
-    "MultiValueTypeInfo",
-    "DMSNodeEntity",
-    "DMSEntity",
+    "CdfResourceEntityList",
+    "ClassEntity",
+    "ClassEntityList",
     "ContainerEntity",
+    "ContainerEntityList",
+    "DMSEntity",
+    "DMSFilter",
+    "DMSNodeEntity",
     "DMSUnknownEntity",
+    "DMSVersionedEntity",
     "DataModelEntity",
+    "EdgeEntity",
+    "Entity",
+    "HasDataFilter",
+    "MultiValueTypeInfo",
+    "NodeTypeFilter",
+    "RawFilter",
     "ReferenceEntity",
-    "ClassEntityList",
-    "CdfResourceEntityList",
-    "ContainerEntityList",
+    "RelationshipEntity",
+    "ReverseConnectionEntity",
+    "T_Entity",
     "URLEntity",
-    "load_value_type",
-    "load_dms_value_type",
-    "load_connection",
+    "Undefined",
+    "UnitEntity",
+    "Unknown",
+    "UnknownEntity",
+    "ViewEntity",
+    "ViewEntityList",
     "WrappedEntity",
-    "NodeTypeFilter",
-    "DMSFilter",
-    "HasDataFilter",
-    "RawFilter",
+    "load_connection",
+    "load_dms_value_type",
+    "load_value_type",
 ]

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

@@ -201,16 +201,36 @@ class Entity(BaseModel, extra="ignore"):
             for field_name, field in self.model_fields.items()
             if (v := getattr(self, field_name)) is not None and field_name not in {"prefix", "suffix"}
         }
+        # We only remove the default values if all the fields are default
+        # For example, if we dump `cdf_cdm:CogniteAsset(version=v1)` and the default is `version=v1`,
+        # we should not remove it unless the space is `cdf_cdm`
+        to_delete: list[str] = []
+        is_removing_defaults = True
         if isinstance(defaults, dict):
             for key, value in defaults.items():
-                if key in model_dump and model_dump[key] == value:
-                    del model_dump[key]
-        # Sorting to ensure deterministic order
-        args = ",".join(f"{k}={v}" for k, v in sorted(model_dump.items(), key=lambda x: x[0]))
-        if self.prefix == Undefined or (isinstance(defaults, dict) and self.prefix == defaults.get("prefix")):
+                if key not in model_dump:
+                    continue
+                if model_dump[key] == value:
+                    to_delete.append(key)
+                elif isinstance(self, EdgeEntity):
+                    # Exception is edge entity, then, we remove all the defaults we can.
+                    continue
+                else:
+                    # Not all fields are default. We should not remove any of them.
+                    is_removing_defaults = False
+                    break
+        if isinstance(defaults, dict) and self.prefix == defaults.get("prefix") and is_removing_defaults:
+            base_id = str(self.suffix)
+        elif self.prefix == Undefined:
             base_id = str(self.suffix)
         else:
+            is_removing_defaults = False
             base_id = f"{self.prefix}:{self.suffix!s}"
+        if is_removing_defaults:
+            for key in to_delete:
+                del model_dump[key]
+        # Sorting to ensure deterministic order
+        args = ",".join(f"{k}={v}" for k, v in sorted(model_dump.items(), key=lambda x: x[0]))
         if args:
             return f"{base_id}({args})"
         else:

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

@@ -8,13 +8,13 @@ from ._rules_input import (
 from ._validation import InformationValidation
 
 __all__ = [
-    "InformationRules",
-    "InformationMetadata",
     "InformationClass",
-    "InformationProperty",
-    "InformationInputRules",
-    "InformationInputMetadata",
     "InformationInputClass",
+    "InformationInputMetadata",
     "InformationInputProperty",
+    "InformationInputRules",
+    "InformationMetadata",
+    "InformationProperty",
+    "InformationRules",
     "InformationValidation",
 ]

cognite/neat/_rules/models/mapping/_classic2core.yaml

@@ -2,10 +2,12 @@ enum:
 - collection: timeseriesType
   description: Time series with double floating point data points.
   name: numeric
+  neatId: http://purl.org/cognite/neat/neatId_d4e42064_7c60_4bfc_b629_a873345f0ee2
   value: numeric
 - collection: timeseriesType
   description: Time series with string data points.
   name: string
+  neatId: http://purl.org/cognite/neat/neatId_8bb46fae_572d_464a_bc61_1055b0aa6c5d
   value: string
 metadata:
   created: '2024-08-29T13:49:56.696000'
@@ -23,6 +25,7 @@ properties:
   description: Description of the instance
   immutable: false
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_d887305d_af9d_4843_af6e_4115a7217d90
   nullable: true
   value_type: text
   view: ClassicAsset
@@ -32,6 +35,7 @@ properties:
   description: Text based labels for generic use, limited to 1000
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_105a0fee_3942_4a6b_9d25_74d46d09c209
   nullable: true
   value_type: text
   view: ClassicAsset
@@ -42,6 +46,7 @@ properties:
   immutable: false
   index: name
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_9559ecd6_a585_4fb9_90b7_7301da5f2170
   nullable: true
   value_type: text
   view: ClassicAsset
@@ -53,6 +58,7 @@ properties:
   immutable: false
   is_list: false
   name: Parent
+  neatId: http://purl.org/cognite/neat/neatId_fdc3280b_0d11_4e66_b2b5_8f37a398bed9
   nullable: true
   value_type: ClassicAsset
   view: ClassicAsset
@@ -64,8 +70,9 @@ properties:
   immutable: false
   index: source
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_247e5b5d_4caf_4312_bd5b_341ed3fc3a03
   nullable: true
-  value_type: cdf_cdm:CogniteSourceSystem(version=v1)
+  value_type: ClassicSourceSystem
   view: ClassicAsset
   view_property: source
 - container: cdf_cdm:CogniteDescribable
@@ -73,6 +80,7 @@ properties:
   description: Alternative names for the node
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_7412eae6_e77c_41b7_92ca_737798f5844a
   nullable: true
   value_type: text
   view: ClassicEvent
@@ -84,6 +92,7 @@ properties:
   immutable: false
   is_list: true
   name: Assets
+  neatId: http://purl.org/cognite/neat/neatId_439538ba_610d_4300_a8cd_aa51dbbf5c1a
   nullable: true
   value_type: ClassicAsset
   view: ClassicEvent
@@ -93,6 +102,7 @@ properties:
   description: Description of the instance
   immutable: false
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_8ab69b21_b84e_4374_a2bd_6b7e6321f22d
   nullable: true
   value_type: text
   view: ClassicEvent
@@ -103,6 +113,7 @@ properties:
   immutable: false
   index: endTime
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_fce69deb_af37_4c49_b1cf_caf569563ba7
   nullable: true
   value_type: timestamp
   view: ClassicEvent
@@ -112,6 +123,7 @@ properties:
   description: Text based labels for generic use, limited to 1000
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_0ed58b00_11ab_4774_9c66_d2eabc5f8f42
   nullable: true
   value_type: text
   view: ClassicEvent
@@ -123,8 +135,9 @@ properties:
   immutable: false
   index: source
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_b0c7d726_6504_4189_896d_1b978d9ef5b2
   nullable: true
-  value_type: cdf_cdm:CogniteSourceSystem(version=v1)
+  value_type: ClassicSourceSystem
   view: ClassicEvent
   view_property: source
 - container: cdf_cdm:CogniteSchedulable
@@ -133,6 +146,7 @@ properties:
   immutable: false
   index: startTime
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_2304763f_d0f3_4723_a4c5_2e879069195d
   nullable: true
   value_type: timestamp
   view: ClassicEvent
@@ -144,6 +158,7 @@ properties:
   immutable: false
   is_list: true
   name: Assets
+  neatId: http://purl.org/cognite/neat/neatId_77ec0da0_839b_4b52_9f64_35382213dac6
   nullable: true
   value_type: ClassicAsset
   view: ClassicFile
@@ -155,6 +170,7 @@ properties:
   immutable: false
   is_list: false
   name: Directory
+  neatId: http://purl.org/cognite/neat/neatId_f06ff883_4f00_4a5d_b9c3_83664b62b573
   nullable: true
   value_type: text
   view: ClassicFile
@@ -164,6 +180,7 @@ properties:
   description: Text based labels for generic use, limited to 1000
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_5d721016_c99e_4f73_8e9d_bdd27b4472c3
   nullable: true
   value_type: text
   view: ClassicFile
@@ -174,6 +191,7 @@ properties:
   immutable: false
   is_list: false
   name: MIME type
+  neatId: http://purl.org/cognite/neat/neatId_d65bd20a_bcb9_4243_b3e9_06188378c4cc
   nullable: true
   value_type: text
   view: ClassicFile
@@ -184,6 +202,7 @@ properties:
   immutable: false
   index: name
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_793d1480_2fba_4dde_8325_c11ef7e57de5
   nullable: true
   value_type: text
   view: ClassicFile
@@ -195,8 +214,9 @@ properties:
   immutable: false
   index: source
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_87aa809b_a7cc_4136_885c_276844ed6b38
   nullable: true
-  value_type: cdf_cdm:CogniteSourceSystem(version=v1)
+  value_type: ClassicSourceSystem
   view: ClassicFile
   view_property: source
 - container: cdf_cdm:CogniteSourceable
@@ -205,6 +225,7 @@ properties:
   immutable: false
   index: sourceCreatedTime
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_26b7346d_3d5f_43a5_894e_96707e89227c
   nullable: true
   value_type: timestamp
   view: ClassicFile
@@ -215,10 +236,20 @@ properties:
   immutable: false
   index: sourceUpdatedTime
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_4b6be7c8_8b26_4f11_9ad5_283123a54c91
   nullable: true
   value_type: timestamp
   view: ClassicFile
   view_property: sourceModifiedTime
+- container: cdf_cdm:CogniteDescribable
+  container_property: name
+  immutable: false
+  is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_7d7c614a_43ea_4fc2_abb0_3f3fdddaa239
+  nullable: true
+  value_type: text
+  view: ClassicSourceSystem
+  view_property: name
 - connection: direct
   container: cdf_cdm:CogniteTimeSeries
   container_property: assets
@@ -226,6 +257,7 @@ properties:
   immutable: false
   is_list: true
   name: Assets
+  neatId: http://purl.org/cognite/neat/neatId_bc4e12b0_82bd_4859_995d_d219181f9b52
   nullable: true
   value_type: ClassicAsset
   view: ClassicTimeSeries
@@ -235,6 +267,7 @@ properties:
   description: Description of the instance
   immutable: false
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_73372a76_aa52_47c1_bdae_34bfbd80f572
   nullable: true
   value_type: text
   view: ClassicTimeSeries
@@ -246,6 +279,7 @@ properties:
   immutable: false
   is_list: false
   name: Is step
+  neatId: http://purl.org/cognite/neat/neatId_cfa476ae_125b_4e6a_8caf_152f44095037
   nullable: false
   value_type: boolean
   view: ClassicTimeSeries
@@ -256,6 +290,7 @@ properties:
   immutable: true
   is_list: false
   name: Type
+  neatId: http://purl.org/cognite/neat/neatId_7943437b_6138_47e6_889a_f414280df48e
   nullable: false
   value_type: enum(collection=timeseriesType)
   view: ClassicTimeSeries
@@ -265,6 +300,7 @@ properties:
   description: Alternative names for the node
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_15cfc454_8fc0_443b_8af6_9db5368d6fd4
   nullable: true
   value_type: text
   view: ClassicTimeSeries
@@ -275,6 +311,7 @@ properties:
   immutable: false
   index: name
   is_list: false
+  neatId: http://purl.org/cognite/neat/neatId_01370dd7_3371_40a1_b4eb_ddfcbbac63c8
   nullable: true
   value_type: text
   view: ClassicTimeSeries
@@ -284,6 +321,7 @@ properties:
   description: Text based labels for generic use, limited to 1000
   immutable: false
   is_list: true
+  neatId: http://purl.org/cognite/neat/neatId_d7e122ee_421a_490f_9bec_b0f9f5ececfd
   nullable: true
   value_type: text
   view: ClassicTimeSeries
@@ -294,6 +332,7 @@ properties:
   immutable: false
   is_list: false
   name: Source unit
+  neatId: http://purl.org/cognite/neat/neatId_3bfc11e0_9d27_47ea_841e_de1e4363dbed
   nullable: true
   value_type: text
   view: ClassicTimeSeries
@@ -305,39 +344,46 @@ properties:
   immutable: false
   is_list: false
   name: Unit
+  neatId: http://purl.org/cognite/neat/neatId_33b75362_e127_4a5a_9690_2df55501250c
   nullable: true
   value_type: cdf_cdm:CogniteUnit(version=v1)
   view: ClassicTimeSeries
   view_property: unitExternalId
 views:
-- description: The CogniteSourceSystem core concept is used to standardize the way
-    source system is stored.
-  implements: cdf_cdm:CogniteDescribable(version=v1)
-  in_model: true
-  view: cdf_cdm:CogniteSourceSystem(version=v1)
 - description: Represents a single unit of measurement
   implements: cdf_cdm:CogniteDescribable(version=v1)
   in_model: true
+  neatId: http://purl.org/cognite/neat/neatId_f3397b4c_b6e0_4701_8ea3_7edbf72bf50d
   view: cdf_cdm:CogniteUnit(version=v1)
 - description: Assets represent systems that support industrial functions or processes.
     Assets are often called 'functional location'.
   implements: cdf_cdm:CogniteAsset(version=v1)
   in_model: true
   name: Asset
+  neatId: http://purl.org/cognite/neat/neatId_23380192_0c8a_4870_9684_cb91e5e5d67d
   view: ClassicAsset
 - description: Represents activities. Activities typically happen over a period and
     have a start and end time.
   implements: cdf_cdm:CogniteActivity(version=v1)
   in_model: true
   name: Activity
+  neatId: http://purl.org/cognite/neat/neatId_3c0bda0e_9af0_4c75_9f1c_d76a6eae2e44
   view: ClassicEvent
 - description: Represents files.
   implements: cdf_cdm:CogniteFile(version=v1)
   in_model: true
   name: File
+  neatId: http://purl.org/cognite/neat/neatId_4a92bfcc_a259_4ec6_b58b_bdb386ba9ce2
   view: ClassicFile
+- description: The CogniteSourceSystem core concept is used to standardize the way
+    source system is stored.
+  implements: cdf_cdm:CogniteSourceSystem(version=v1)
+  in_model: true
+  neatId: http://purl.org/cognite/neat/neatId_280264e2_a809_4feb_81e3_454adc00fdde
+  view: ClassicSourceSystem
 - description: Represents a series of data points in time order.
   implements: cdf_cdm:CogniteTimeSeries(version=v1)
   in_model: true
   name: Time series
+  neatId: http://purl.org/cognite/neat/neatId_0730d4f7_51ce_40de_a8af_b9efff498b06
   view: ClassicTimeSeries

cognite/neat/_rules/transformers/__init__.py

@@ -14,21 +14,21 @@ from ._pipelines import ImporterPipeline
 from ._verification import VerifyAnyRules, VerifyDMSRules, VerifyInformationRules
 
 __all__ = [
-    "ImporterPipeline",
-    "RulesTransformer",
-    "RulesPipeline",
-    "InformationToDMS",
+    "AsParentPropertyId",
     "ConvertToRules",
-    "PrefixEntities",
     "DMSToInformation",
-    "VerifyDMSRules",
-    "VerifyInformationRules",
-    "VerifyAnyRules",
+    "ImporterPipeline",
+    "InformationToDMS",
     "MapOneToOne",
-    "ToCompliantEntities",
-    "RuleMapper",
-    "ToExtension",
+    "PrefixEntities",
     "ReduceCogniteModel",
+    "RuleMapper",
+    "RulesPipeline",
+    "RulesTransformer",
     "SetIDDMSModel",
-    "AsParentPropertyId",
+    "ToCompliantEntities",
+    "ToExtension",
+    "VerifyAnyRules",
+    "VerifyDMSRules",
+    "VerifyInformationRules",
 ]

cognite/neat/_rules/transformers/_pipelines.py

@@ -1,28 +1,33 @@
 from collections.abc import Iterable
+from typing import TYPE_CHECKING
 
 from cognite.neat._issues.errors import NeatValueError
 from cognite.neat._rules._shared import InputRules, MaybeRules, VerifiedRules
-from cognite.neat._rules.importers import BaseImporter
 from cognite.neat._rules.models import VERIFIED_RULES_BY_ROLE, RoleTypes
 
 from ._base import RulesPipeline, RulesTransformer
 from ._converters import ConvertToRules
 from ._verification import VerifyAnyRules
 
+if TYPE_CHECKING:
+    from cognite.neat._rules.importers import BaseImporter
+
 
 class ImporterPipeline(RulesPipeline[InputRules, VerifiedRules]):
     """This is a standard pipeline that verifies, convert and return the rules from the importer."""
 
     def __init__(
         self,
-        importer: BaseImporter[InputRules],
+        importer: "BaseImporter[InputRules]",
         items: Iterable[RulesTransformer[InputRules, VerifiedRules]] | None = None,
     ) -> None:
         super().__init__(items or [])
         self._importer = importer
 
     @classmethod
-    def _create_pipeline(cls, importer: BaseImporter[InputRules], role: RoleTypes | None = None) -> "ImporterPipeline":
+    def _create_pipeline(
+        cls, importer: "BaseImporter[InputRules]", role: RoleTypes | None = None
+    ) -> "ImporterPipeline":
         items: list[RulesTransformer] = [VerifyAnyRules(errors="continue", validate=True)]
         if role is not None:
             out_cls = VERIFIED_RULES_BY_ROLE[role]
@@ -30,7 +35,7 @@ class ImporterPipeline(RulesPipeline[InputRules, VerifiedRules]):
         return cls(importer, items)
 
     @classmethod
-    def try_verify(cls, importer: BaseImporter, role: RoleTypes | None = None) -> MaybeRules[VerifiedRules]:
+    def try_verify(cls, importer: "BaseImporter", role: RoleTypes | None = None) -> MaybeRules[VerifiedRules]:
         """This is a standard pipeline that verifies, convert and return the rules from the importer.
 
         Args:
@@ -43,7 +48,7 @@ class ImporterPipeline(RulesPipeline[InputRules, VerifiedRules]):
         return cls._create_pipeline(importer, role).try_execute()
 
     @classmethod
-    def verify(cls, importer: BaseImporter, role: RoleTypes | None = None) -> VerifiedRules:
+    def verify(cls, importer: "BaseImporter", role: RoleTypes | None = None) -> VerifiedRules:
         """Verify the rules."""
         return cls._create_pipeline(importer, role).execute()