[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pyg-team · Sep 6, 2024 · 15aff00 · 15aff00
1 parent c16f7d5
commit 15aff00
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/test/data/test_multi_nested_tensor.py b/test/data/test_multi_nested_tensor.py
@@ -33,13 +33,11 @@ def column_select(
 
 
 def test_is_floating_point():
-    met = MultiNestedTensor.from_tensor_mat([
-        [torch.tensor([1], dtype=torch.long)]
-    ])
+    met = MultiNestedTensor.from_tensor_mat(
+        [[torch.tensor([1], dtype=torch.long)]])
     assert not met.is_floating_point()
-    met = MultiNestedTensor.from_tensor_mat([
-        [torch.tensor([1], dtype=torch.float32)]
-    ])
+    met = MultiNestedTensor.from_tensor_mat(
+        [[torch.tensor([1], dtype=torch.float32)]])
     assert met.is_floating_point()