From e58189e5f06eeef91fbb9bac88762f105d87bbee Mon Sep 17 00:00:00 2001 From: zer0int <132047210+zer0int@users.noreply.github.com> Date: Tue, 27 Aug 2024 22:43:48 +0200 Subject: [PATCH] Fix generator expression error w/ list comprehension (#7) * Fix incompatibility with older Python versions in expression flip function handler --- py/expression_handlers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/py/expression_handlers.py b/py/expression_handlers.py index 2663942..d669c0a 100644 --- a/py/expression_handlers.py +++ b/py/expression_handlers.py @@ -102,11 +102,11 @@ class FlipHandler(NormHandler): return torch.flip(tensor, (dim,)) result = tensor.detach().clone() pivot = tensor.shape[dim] // 2 - out_slice = ( + out_slice = tuple( np.s_[:] if d != dim else np.s_[pivot:] for d in range(tensor.ndim) ) - in_slice = (np.s_[:] if d != dim else np.s_[:pivot] for d in range(tensor.ndim)) - result[*out_slice] = torch.flip(tensor[*in_slice], dims=(dim,)) + in_slice = tuple(np.s_[:] if d != dim else np.s_[:pivot] for d in range(tensor.ndim)) + result[out_slice] = torch.flip(tensor[in_slice], dims=(dim,)) return result