Remove unnecessary main function in test

Summary: `main()` function are not needed because test discovery should detect those automatically. Reviewed By: hjmshi Differential Revision: D52412607 fbshipit-source-id: 3e87b084a36c82f443ec7c67d252351af5a898e1
facebookresearch · Jan 2, 2024 · a199f27 · a199f27
1 parent b659e78
commit a199f27
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/distributed_shampoo/tests/shampoo_model_utils_test.py b/distributed_shampoo/tests/shampoo_model_utils_test.py
@@ -135,7 +135,3 @@ def test_initialization(self):
                     torch.testing.assert_close(
                         original_linear.weight, combined_linear.combined_weight
                     )
-
-
-if __name__ == "__main__":
-    unittest.main()