diff --git a/cognite/client/data_classes/relationships.py b/cognite/client/data_classes/relationships.py index eb600f9659..1e8a089559 100644 --- 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