From cc4cb6b2678f7279d037b63c0957f73fd30ae2e4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Mon, 8 Jun 2020 19:30:33 +0200 Subject: [PATCH] prevent side effects in tests --- tests/core/test_decorators.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/core/test_decorators.py b/tests/core/test_decorators.py index e23db3d74c97c..0f35a1630e1d9 100644 --- a/tests/core/test_decorators.py +++ b/tests/core/test_decorators.py @@ -15,10 +15,13 @@ def test_auto_move_data(src_device, dest_device): """ Test that the decorator moves the data to the device the model is on. """ + class CurrentModel(EvalModelTemplate): + pass + # apply the decorator - EvalModelTemplate.forward = auto_move_data(EvalModelTemplate.forward) + CurrentModel.forward = auto_move_data(CurrentModel.forward) - model = EvalModelTemplate() + model = CurrentModel() model = model.to(dest_device) model.prepare_data() loader = model.train_dataloader()