From 749f0a780fece177623341f83a68dde9da2161ef Mon Sep 17 00:00:00 2001 From: svij Date: Mon, 4 May 2026 19:10:31 +0000 Subject: [PATCH] fix --- gigl/distributed/dataset_factory.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gigl/distributed/dataset_factory.py b/gigl/distributed/dataset_factory.py index 1f78f83f3..61a08f051 100644 --- a/gigl/distributed/dataset_factory.py +++ b/gigl/distributed/dataset_factory.py @@ -624,6 +624,7 @@ def build_dataset_from_task_config_uri( serialized_graph_metadata=serialized_graph_metadata, sample_edge_direction=sample_edge_direction, distributed_context=distributed_context, + should_load_tensors_in_parallel=should_load_tensors_in_parallel, partitioner_class=partitioner_class, splitter=splitter, _ssl_positive_label_percentage=ssl_positive_label_percentage,