From 470ea21eebd2aed5ffbab1550cce8ebcf5daed90 Mon Sep 17 00:00:00 2001 From: SkandanC Date: Sun, 3 Jul 2022 23:26:32 -0400 Subject: [PATCH] Fix type conversion and ValueError --- SiPANN/nn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/SiPANN/nn.py b/SiPANN/nn.py index 1a7b2d9..87e8183 100644 --- a/SiPANN/nn.py +++ b/SiPANN/nn.py @@ -69,9 +69,10 @@ def cartesian_product(arrays): la = len(arrays) dtype = np.find_common_type([a.dtype for a in arrays], []) - l = [float(_a) for _a in arrays] + l = [_a.astype(float) for _a in arrays] arr = np.empty([a.size for a in arrays] + [la], dtype=dtype) for i, a in enumerate(l): + a = a.reshape(a.shape + (1,) * (la - 1)) arr[..., i] = a return arr.reshape(-1, la)