From a6209cba5943690d711a427e2a8043ec05896d4a Mon Sep 17 00:00:00 2001
From: gcroci2 <crocioni.giulia@gmail.com>
Date: Fri, 20 Oct 2023 14:25:17 +0200
Subject: [PATCH] fix test_optim

---
 tests/test_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_trainer.py b/tests/test_trainer.py
index 033f8a934..4ebbca2e5 100644
--- a/tests/test_trainer.py
+++ b/tests/test_trainer.py
@@ -483,7 +483,7 @@ def test_optim(self):
                 dataset_test=dataset,
                 pretrained_model=self.save_path)
 
-        assert isinstance(trainer_pretrained.optimizer, optimizer)
+        assert str(type(trainer_pretrained.optimizer)) == "<class 'torch.optim.adamax.Adamax'>"
         assert trainer_pretrained.lr == lr
         assert trainer_pretrained.weight_decay == weight_decay