From 935ffddd6c4c4120e10d679074d7903c56789c16 Mon Sep 17 00:00:00 2001 From: Lars Kuehmichel Date: Fri, 27 Oct 2023 16:24:44 +0200 Subject: [PATCH] Fix test for lightning 2.1 --- tests/module_tests/test_trainable.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/module_tests/test_trainable.py b/tests/module_tests/test_trainable.py index a7bb80c..31283cf 100644 --- a/tests/module_tests/test_trainable.py +++ b/tests/module_tests/test_trainable.py @@ -132,7 +132,7 @@ def test_continue_training(dummy_model): print("Finished Training. Loading Checkpoint.") checkpoint = find_checkpoint() - trained_model = dummy_model.load_from_checkpoint(checkpoint) + trained_model = dummy_model.__class__.load_from_checkpoint(checkpoint) print("Continuing Training.") trained_model.fit(