diff --git a/diffusion_policy/model/common/rotation_transformer.py b/diffusion_policy/model/common/rotation_transformer.py index 0697d981..a801d285 100644 --- a/diffusion_policy/model/common/rotation_transformer.py +++ b/diffusion_policy/model/common/rotation_transformer.py @@ -40,7 +40,7 @@ def __init__(self, getattr(pt, f'matrix_to_{from_rep}') ] if from_convention is not None: - funcs = [functools.partial(func, convernsion=from_convention) + funcs = [functools.partial(func, convention=from_convention) for func in funcs] forward_funcs.append(funcs[0]) inverse_funcs.append(funcs[1]) @@ -51,7 +51,7 @@ def __init__(self, getattr(pt, f'{to_rep}_to_matrix') ] if to_convention is not None: - funcs = [functools.partial(func, convernsion=to_convention) + funcs = [functools.partial(func, convention=to_convention) for func in funcs] forward_funcs.append(funcs[0]) inverse_funcs.append(funcs[1])