From d9e829672718a266fcbd01380e66143f02a7f30c Mon Sep 17 00:00:00 2001 From: Ananth Subramaniam Date: Thu, 21 Nov 2024 13:37:06 -0800 Subject: [PATCH] add deprecation warning for old preemption callback Signed-off-by: Ananth Subramaniam --- nemo/utils/callbacks/preemption.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nemo/utils/callbacks/preemption.py b/nemo/utils/callbacks/preemption.py index 178fe94cee7c1..eb8b70daf60d7 100644 --- a/nemo/utils/callbacks/preemption.py +++ b/nemo/utils/callbacks/preemption.py @@ -19,8 +19,10 @@ from lightning.pytorch.callbacks import Callback from nemo.utils import logging +from nemo.utils.decorators import deprecated +@deprecated(explanation="nemo.utils.callbacks.preemption.PreemptionCallback is deprecated in favor of nemo.lightning.pytorch.callbacks.preemption.PreemptionCallback") class PreemptionCallback(Callback): """ PreemptionCallback class creates a callback that checks for preemption during training at the end of every step.