Skip to content

Commit d1a2c88

Browse files
committed
E226 missing whitespace around arithmetic operator fix
1 parent e97d792 commit d1a2c88

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

torchsparsegradutils/tests/test_sparse_matmul.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -279,9 +279,9 @@ def test_sparse_mm_memory_advantage(device, value_dtype, index_dtype, mem_shapes
279279
# # prints for debugging
280280
# print(
281281
# f"\nLayout={layout}, "
282-
# f"torch sparse.mm={mem_torch_sparse_mm/1e6:.1f}MB, "
283-
# f"tsgu sparse_mm={mem_tsgu_sparse_mm/1e6:.1f}MB, "
284-
# f"dense={mem_torch_dense_mm/1e6:.1f}MB"
282+
# f"torch sparse.mm={mem_torch_sparse_mm / 1e6:.1f}MB, "
283+
# f"tsgu sparse_mm={mem_tsgu_sparse_mm / 1e6:.1f}MB, "
284+
# f"dense={mem_torch_dense_mm / 1e6:.1f}MB"
285285
# )
286286

287287
# sanity: we still got sparse grads
@@ -290,7 +290,7 @@ def test_sparse_mm_memory_advantage(device, value_dtype, index_dtype, mem_shapes
290290
# confirm memory saving
291291
assert mem_tsgu_sparse_mm < mem_torch_sparse_mm, (
292292
f"sparse_mm should use less GPU memory than torch.sparse.mm "
293-
f"({mem_tsgu_sparse_mm/1e6:.1f}MB vs {mem_torch_sparse_mm/1e6:.1f}MB)"
293+
f"({mem_tsgu_sparse_mm / 1e6:.1f}MB vs {mem_torch_sparse_mm / 1e6:.1f}MB)"
294294
)
295295
# NOTE: This test only passes for sparse COO
296296

0 commit comments

Comments
 (0)