From 72ede388cbba22bc3599ded6135ff060cef0b113 Mon Sep 17 00:00:00 2001 From: Miguel de la Varga Date: Sun, 26 Oct 2025 12:23:14 +0100 Subject: [PATCH] Add TypeError handling to `_concatenate` for unsupported tensor types --- gempy_engine/core/backend_tensor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gempy_engine/core/backend_tensor.py b/gempy_engine/core/backend_tensor.py index 1955b79..8f96523 100644 --- a/gempy_engine/core/backend_tensor.py +++ b/gempy_engine/core/backend_tensor.py @@ -197,10 +197,11 @@ def _array(array_like, dtype=None): def _concatenate(tensors, axis=0, dtype=None): # Switch if tensor is numpy array or a torch tensor match type(tensors[0]): - case numpy.ndarray: + case _ if any(isinstance(t, numpy.ndarray) for t in tensors): return numpy.concatenate(tensors, axis=axis) case torch.Tensor: return torch.cat(tensors, dim=axis) + raise TypeError("Unsupported tensor type") def _transpose(tensor, axes=None): return tensor.transpose(axes[0], axes[1])