Roll forward PR #104

google-deepmind · Jul 10, 2024 · ea337b4 · ea337b4
1 parent 5631f12
commit ea337b4
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/clrs/_src/probing.py b/clrs/_src/probing.py
@@ -233,7 +233,6 @@ def heap(A_pos: np.ndarray, heap_size: int) -> np.ndarray:
 
 def graph(A: np.ndarray) -> np.ndarray:
   """Constructs a `graph` probe."""
-  probe = (A != 0) * 1.0
   probe = ((A + np.eye(A.shape[0])) != 0) * 1.0
   return probe