From a199f27daf3d45a63f2dd2d03754015fa9c6e3fa Mon Sep 17 00:00:00 2001 From: Tsung-Hsien Lee Date: Tue, 2 Jan 2024 15:36:51 -0800 Subject: [PATCH] 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 --- distributed_shampoo/tests/shampoo_model_utils_test.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/distributed_shampoo/tests/shampoo_model_utils_test.py b/distributed_shampoo/tests/shampoo_model_utils_test.py index 88c9afd..bf477b7 100644 --- 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()