fix: Loading with cognite client

cognitedata · Nov 15, 2023 · cad6dbd · cad6dbd
1 parent 5f5ab9e
commit cad6dbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cognite/client/data_classes/relationships.py b/cognite/client/data_classes/relationships.py
@@ -95,9 +95,9 @@ def _validate_resource_type(self, resource_type: str | None) -> None:
     def _load(cls, resource: dict, cognite_client: CogniteClient | None = None) -> Relationship:
         instance = super()._load(resource, cognite_client)
         if instance.source is not None:
-            instance.source = instance._convert_resource(instance.source, instance.source_type)  # type: ignore
+            instance.source = instance._convert_resource(instance.source, instance.source_type, cognite_client)  # type: ignore
         if instance.target is not None:
-            instance.target = instance._convert_resource(instance.target, instance.target_type)  # type: ignore
+            instance.target = instance._convert_resource(instance.target, instance.target_type, cognite_client)  # type: ignore
         instance.labels = Label._load_list(instance.labels)
         return instance