From 846ece46d3b3604ca53c26ef8f63f6e0fc82c2a7 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 e23db3d74c97ca..0f35a1630e1d96 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()