Fix disjoint mode to only operate in temporal sampling cases (#8482)

pyg-team · Nov 29, 2023 · 1d11179 · 1d11179
1 parent 12a2fb2
commit 1d11179
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_geometric/distributed/dist_neighbor_sampler.py b/torch_geometric/distributed/dist_neighbor_sampler.py
@@ -619,12 +619,12 @@ def _sample_one_hop(
             True,  # csc
             self.replace,
             self.subgraph_type != SubgraphType.induced,
-            self.disjoint or self.time_attr is not None,
+            self.disjoint and self.time_attr is not None,
             self.temporal_strategy,
         )
         node, edge, cumsum_neighbors_per_node = out
 
-        if self.disjoint or self.time_attr is not None:
+        if self.disjoint and self.time_attr is not None:
             # We create a batch during the step of merging sampler outputs.
             _, node = node.t().contiguous()