fix test

Lightning-AI · May 15, 2020 · 18dff98 · 18dff98
1 parent 8779ba5
commit 18dff98
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_hooks.py b/tests/models/test_hooks.py
@@ -49,7 +49,7 @@ def transfer_batch_to_device(self, batch, device):
                 batch.targets = batch.targets.to(device)
             return batch
 
-    model = CurrentTestModel(tutils.get_default_hparams())
+    model = CurrentTestModel()
     batch = CustomBatch((torch.zeros(5, 28), torch.ones(5, 1, dtype=torch.long)))
 
     trainer = Trainer()