[Train] Backwards Compatibility for TrainingCallback (#19566)

ray-project · Oct 21, 2021 · 8792714 · 8792714
1 parent 8acece8
commit 8792714
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/python/ray/util/sgd/v2/__init__.py b/python/ray/util/sgd/v2/__init__.py
@@ -1,4 +1,8 @@
-from ray.train import *  # noqa: F401, F403
 import warnings
 
+from ray.train import *  # noqa: F401, F403
+from ray.train import TrainingCallback
+
+SGDCallback = TrainingCallback
+
 warnings.warn("ray.util.sgd.v2 has been moved to ray.train.")
diff --git a/python/ray/util/sgd/v2/callbacks/__init__.py b/python/ray/util/sgd/v2/callbacks/__init__.py
@@ -1 +1,4 @@
 from ray.train.callbacks import *  # noqa: F401, F403
+from ray.train.callbacks import TrainingCallback
+
+SGDCallback = TrainingCallback