From 4a9ddd540fb10485d5d63c0357dcb9f50a890e90 Mon Sep 17 00:00:00 2001
From: Stanislaw Adaszewski <stanislaw.adaszewski@roche.com>
Date: Fri, 17 Jul 2020 13:27:30 +0200
Subject: [PATCH] Fix in test_decode_layer_05().

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

diff --git a/tests/icosagon/test_declayer.py b/tests/icosagon/test_declayer.py
index 732fd45..014d304 100644
--- a/tests/icosagon/test_declayer.py
+++ b/tests/icosagon/test_declayer.py
@@ -166,7 +166,7 @@ def test_decode_layer_05():
     # print(rel_pred.edges_neg.train)
 
     repr_in = in_layer(None)
-    assert isinstance(repr_in, list)
+    assert isinstance(repr_in, torch.nn.ParameterList)
     assert len(repr_in) == 1
     assert isinstance(repr_in[0], torch.Tensor)
     assert torch.all(repr_in[0].to_dense() == torch.eye(10))