[release] fix hvd test. (ray-project#32558)

Signed-off-by: xwjiang2010 <[email protected]>
Ransnomy · Feb 15, 2023 · 43dd0dc · 43dd0dc
1 parent 789e683
commit 43dd0dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/release/train_tests/horovod/train_horovod_multi_node_test.py b/release/train_tests/horovod/train_horovod_multi_node_test.py
@@ -7,7 +7,7 @@
 from ray.train.examples.horovod.horovod_example import (
     train_func as horovod_torch_train_func,
 )
-from ray.train.constants import TRAINING_ITERATION
+from ray.tune.result import TRAINING_ITERATION
 from ray.train.horovod.horovod_trainer import HorovodTrainer
 
 if __name__ == "__main__":