From 624670bf65414973008a22e2d9ed1a664be1fa75 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Tue, 16 Jan 2024 02:11:34 +0000
Subject: [PATCH] Fix type bug in test_tensor.py
---
aidge_backend_cpu/unit_tests/test_tensor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_backend_cpu/unit_tests/test_tensor.py b/aidge_backend_cpu/unit_tests/test_tensor.py
index 830b5cb5..37531b43 100644
--- a/aidge_backend_cpu/unit_tests/test_tensor.py
+++ b/aidge_backend_cpu/unit_tests/test_tensor.py
@@ -59,7 +59,7 @@ class test_tensor(unittest.TestCase):
def test_get_set(self):
dims = [2,2,2]
- np_array = np.arange(8).reshape(dims)
+ np_array = np.arange(8).reshape(dims).astype(np.int32)
# Numpy -> Tensor
t = aidge_core.Tensor(np_array)
for i in range(8):
--
GitLab