linting

dmlc · Jul 18, 2024 · 9f40899 · 9f40899
1 parent 10ba72c
commit 9f40899
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py
@@ -95,7 +95,7 @@ def read_async(self, ids: torch.Tensor):
         for _ in range(
             self._fallback_feature.read_async_num_stages(missing_keys.device)
         ):
-            missing_values_future = next(fallback_reader)
+            missing_values_future = next(fallback_reader, None)
             yield  # fallback feature stages.
         values[missing_index] = missing_values_future.wait()