Skip to content

Commit

Permalink
Merge pull request #307 from Visual-Behavior/issue306_cat_augmentedte…
Browse files Browse the repository at this point in the history
…nsor_fix

fix cat for tuple of augmented tensor
  • Loading branch information
Data-Iab authored Nov 24, 2022
2 parents fc1aa9e + d7870a8 commit 3787391
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aloscene/tensors/augmented_tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 3787391

Please sign in to comment.