diff --git a/src/finn/custom_op/fpgadataflow/lookup.py b/src/finn/custom_op/fpgadataflow/lookup.py index e7a957169f40ebd855ece1d380290851cedd86d6..6a7848ccda6915836d183ff324067f4e19b21ec3 100644 --- a/src/finn/custom_op/fpgadataflow/lookup.py +++ b/src/finn/custom_op/fpgadataflow/lookup.py @@ -79,10 +79,10 @@ class Lookup(HLSCustomOp): return self.get_normal_output_shape() def make_shape_compatible_op(self, model): - exp_ishape = self.get_normal_input_shape() - oshape = self.get_normal_output_shape() + exp_ishape = tuple(self.get_normal_input_shape()) + oshape = tuple(self.get_normal_output_shape()) ishape = tuple(model.get_tensor_shape(self.onnx_node.input[0])) - assert ishape == exp_ishape, "Unexpect input shape for Lookup: %s vs %s" % ( + assert ishape == exp_ishape, "Unexpected input shape for Lookup: %s vs %s" % ( str(exp_ishape), str(ishape), )