Skip to content

Commit a2fa7c0

Browse files
committed
Add fn
1 parent f3dc154 commit a2fa7c0

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/trainer/optimization/test_manual_optimization.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,11 @@ def training_step(self, batch, batch_idx):
197197
# avoid returning a value
198198
super().training_step(batch, batch_idx)
199199

200+
def training_epoch_end(self, outputs) -> None:
201+
# outputs is empty as training_step does not return
202+
# and it is not automatic optimization
203+
assert len(outputs) == 0
204+
200205
model = TestModel()
201206
model.val_dataloader = None
202207

0 commit comments

Comments
 (0)