diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py index 9b158d73b..d42c2fca1 100755 --- a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py +++ b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py @@ -82,7 +82,7 @@ def _prepare_data(self, data): if len(data) == 2: return data - return numpy.array([numpy.asarray(data[:, :-1]), numpy.asarray([[1, 0] if label == 0 else [0, 1] for label in data[:, -1]])]) + return numpy.array([numpy.array(data[:, :-1]), numpy.array([[1, 0] if label == 0 else [0, 1] for label in data[:, -1]])]) def calculate_metrics(y_pred, y_truth, metrics):