From 658cded5714e806e7432d984c31bbf3d4acbc20c Mon Sep 17 00:00:00 2001 From: runame Date: Thu, 14 Nov 2024 02:02:38 +0000 Subject: [PATCH] Add -I flag to unittest to avoid using local modules --- makefile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/makefile b/makefile index bfc8d1f..2c4956b 100644 --- a/makefile +++ b/makefile @@ -9,8 +9,8 @@ type-check: @mypy . test: - @python3 -m unittest discover -s tests/ -p "*_test.py" - @python3 -m unittest discover -s distributed_shampoo/tests/ -p "*_test.py" + @python3 -I -m unittest discover -s tests/ -p "*_test.py" + @python3 -I -m unittest discover -s distributed_shampoo/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