Fix int fill-value error in test_modules.py

2020-07-30 11:42:37 +02:00 · 2020-07-30 11:42:37 +02:00 · b138277608
commit b138277608
parent 9ccbec52f7
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modules.py
+++ b/tests/test_modules.py
@ -199,7 +199,7 @@ class TestPrototypes(unittest.TestCase):

    def test_prototypes1d_func_initializer(self):
        def my_initializer(*args, **kwargs):
-            return torch.full((2, 99), 99), torch.tensor([0, 1])
+            return torch.full((2, 99), 99.0), torch.tensor([0, 1])

        p1 = prototypes.Prototypes1D(input_dim=99,
                                     nclasses=2,