Automated Code Change

PiperOrigin-RevId: 696571638
tensorflow · Nov 14, 2024 · 2c817c1 · 2c817c1
1 parent 93cd2be
commit 2c817c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_gnn/sampler/unsupported/sampling_lib.py b/tensorflow_gnn/sampler/unsupported/sampling_lib.py
@@ -103,7 +103,7 @@ def random_weighted_fn(edge: Edge) -> float:
 
 def is_deterministic(sampling_op: sampling_spec_pb2.SamplingOp) -> bool:
   """Returns `True` if sampling operation produces resample_for_each_path results."""
-  return _DETERMINISTIC_SAMPLING_STRATEGIES.get(sampling_op.strategy)
+  return _DETERMINISTIC_SAMPLING_STRATEGIES.get(sampling_op.strategy)  # pytype: disable=bad-return-type
 
 
 # Sampling frontier is a mapping from (sample id, node id) pair to the number