From d7870a849e2d79307020da0322288b090ea5cd20 Mon Sep 17 00:00:00 2001 From: Julien Salotti Date: Mon, 21 Nov 2022 13:11:59 +0100 Subject: [PATCH] fix cat for tuple of augmented tensor --- aloscene/tensors/augmented_tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aloscene/tensors/augmented_tensor.py b/aloscene/tensors/augmented_tensor.py index 255f27e1..2bc76c7a 100644 --- a/aloscene/tensors/augmented_tensor.py +++ b/aloscene/tensors/augmented_tensor.py @@ -571,7 +571,7 @@ def __torch_function__(cls, func, types, args=(), kwargs=None): self = _torch_function_get_self(cls, func, types, args, kwargs) def _merging_frame(args): - if len(args) >= 1 and isinstance(args[0], list): + if len(args) >= 1 and isinstance(args[0], (list, tuple)): for el in args[0]: if isinstance(el, cls): return True