Fix numpy upgrade to 2.0.0 BUFSIZE import error

microsoft · Jun 18, 2024 · f38250e · f38250e
1 parent b33873d
commit f38250e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/autotuning/scheduler.py b/deepspeed/autotuning/scheduler.py
@@ -5,7 +5,7 @@
 
 import copy
 
-from numpy import BUFSIZE
+from numpy._core.umath import BUFSIZE
 import json
 import subprocess
 import sys