Skip to content

Commit 498e981

Browse files
committed
undo app change
1 parent 63de8bf commit 498e981

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/app_components/python/component_tracer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def on_train_start(self, trainer, pl_module) -> None:
2727
print("Even the Lightning Work is available and state transfer works !")
2828
print(self.lightning_work)
2929

30-
def on_train_batch_end(self, trainer, *_) -> None:
30+
def on_batch_end(self, trainer, *_) -> None:
3131
# On every batch end, collects some information.
3232
# This is communicated automatically to the rest of the app,
3333
# so you can track your training in real time in the Lightning App UI.

0 commit comments

Comments
 (0)