style

borisfom · borisfom · commit 87bef8e38e1d · 2025-01-13T11:26:04.000-08:00
Signed-off-by: Boris Fomitchev &lt;bfomitchev@nvidia.com&gt;
diff --git a/cuequivariance_torch/tests/primitives/primitive_export_test.py b/cuequivariance_torch/tests/primitives/primitive_export_test.py
@@ -74,7 +74,6 @@ def test_script_fused_tp_3(mode, tmp_path):
     torch.testing.assert_close(out11, out22)
 
 
-
 @pytest.mark.parametrize("mode", export_modes)
 def test_script_fused_tp_4(mode, tmp_path):
     if not torch.cuda.is_available():
diff --git a/cuequivariance_torch/tests/primitives/tensor_product_test.py b/cuequivariance_torch/tests/primitives/tensor_product_test.py
@@ -89,6 +89,7 @@ def make_descriptors():
         (torch.bfloat16, torch.float32, 1.0),
     ]
 
+
 @pytest.mark.parametrize("batch_size", [0, 3])
 @pytest.mark.parametrize("use_fallback", [True, False])
 @pytest.mark.parametrize("dtype, math_dtype, tol", settings)

Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,7 @@ def make_descriptors():`
`89`	`89`	`(torch.bfloat16, torch.float32, 1.0),`
`90`	`90`	`]`
`91`	`91`
	`92`	`+`
`92`	`93`	`@pytest.mark.parametrize("batch_size", [0, 3])`
`93`	`94`	`@pytest.mark.parametrize("use_fallback", [True, False])`
`94`	`95`	`@pytest.mark.parametrize("dtype, math_dtype, tol", settings)`