Update tests/unit/alexnet_model.py

microsoft · Dec 14, 2023 · 382a4ed · 382a4ed
1 parent bf62691
commit 382a4ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit/alexnet_model.py b/tests/unit/alexnet_model.py
@@ -10,7 +10,8 @@
 import torch.nn.functional as F
 import deepspeed
 import deepspeed.comm as dist
-import deepspeed.runtime.utils as ds_utils, required_torch_version
+import deepspeed.runtime.utils as ds_utils
+from deepspeed.runtime.utils import required_torch_version
 from deepspeed.accelerator import get_accelerator
 from deepspeed.runtime.pipe.module import PipelineModule, LayerSpec