From 7dd5b411a33832bc94c665b71730b99fd34af230 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 7 Feb 2020 20:19:57 +0100
Subject: [PATCH] [Util] more relaxation for ndarray type detection
---
src/finn/util/data_packing.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/finn/util/data_packing.py b/src/finn/util/data_packing.py
index 551849558..31ad66868 100644
--- a/src/finn/util/data_packing.py
+++ b/src/finn/util/data_packing.py
@@ -280,7 +280,9 @@ def packed_bytearray_to_finnpy(packed_bytearray, dtype, output_shape=None):
given DataType. output_shape can be specified to remove padding from the
packed dimension, or set to None to be inferred from the input."""
- if type(packed_bytearray) != np.ndarray or packed_bytearray.dtype != np.uint8:
+ if (
+ not issubclass(type(packed_bytearray), np.ndarray)
+ ) or packed_bytearray.dtype != np.uint8:
raise Exception("packed_bytearray_to_finnpy needs NumPy uint8 arrays")
if packed_bytearray.ndim == 0:
raise Exception("packed_bytearray_to_finnpy expects at least 1D ndarray")
--
GitLab