From 239c3429dd6e1d24a7d740328c6648f592df30cf Mon Sep 17 00:00:00 2001 From: runame Date: Thu, 14 Nov 2024 01:58:11 +0000 Subject: [PATCH] Next try --- makefile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/makefile b/makefile index 7fdd64f..2c4956b 100644 --- a/makefile +++ b/makefile @@ -11,7 +11,7 @@ type-check: test: @python3 -I -m unittest discover -s tests/ -p "*_test.py" @python3 -I -m unittest discover -s distributed_shampoo/tests/ -p "*_test.py" - @python3 -I -m unittest discover -s distributed_shampoo/utils/tests/ -p "*_test.py" + @python3 -m unittest discover -s distributed_shampoo/utils/tests/ -p "*_test.py" @python3 -m unittest discover -s distributed_shampoo/gpu_tests/ -p "*_test.py" @python3 -m unittest distributed_shampoo/utils/gpu_tests/shampoo_dist_utils_test.py @torchrun --standalone --nnodes=1 --nproc_per_node=2 -m unittest distributed_shampoo/utils/gpu_tests/shampoo_ddp_distributor_test.py