test: Increase threshold of val loss in fit test

lRomul · Sep 8, 2020 · 75c9457 · 75c9457
1 parent 54136e6
commit 75c9457
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,5 +1,4 @@
 import pytest
-import logging
 
 import torch
 from torch import nn

diff --git a/tests/engine/test_engine.py b/tests/engine/test_engine.py
@@ -15,7 +15,8 @@ def test_state_update():
 
 class TestEngineMethods:
     def test_add_event_handler(self):
-        def some_function(): pass
+        def some_function():
+            pass
         engine = Engine(some_function)
         assert len(engine.event_handlers[Events.START]) == 0
         engine.add_event_handler(Events.START, some_function)

diff --git a/tests/model/test_model.py b/tests/model/test_model.py
@@ -79,7 +79,7 @@ def test_fit_train_val_loader(self,
                   num_epochs=32)
         val_loss_after = model.validate(val_loader)['val_loss']
         assert val_loss_after < val_loss_before
-        assert val_loss_after < 0.1
+        assert val_loss_after < 0.3
 
     def test_save(self, tmpdir, linear_argus_model_instance):
         path = str(tmpdir.mkdir("experiment").join("model.pth"))