don't use __main__ for test

axolotl-ai-cloud · Sep 1, 2024 · 2f58ea3 · 2f58ea3
1 parent dff66e8
commit 2f58ea3
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/tests/e2e/integrations/liger.py b/tests/e2e/integrations/liger.py
@@ -108,7 +108,3 @@ def test_llama_w_flce(self, temp_dir):
 
         train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta)
         assert (Path(temp_dir) / "model.safetensors").exists()
-
-
-if __name__ == "__main__":
-    unittest.main()