hongshaorou/diffusion_policy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix typo in rotation_transformer.py

Browse Source
This commit is contained in:
Dian Wang 2023-10-25 10:43:25 -04:00 committed by GitHub
parent 7dd9dc417a
commit de4384e84a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
diffusion_policy/model/common/rotation_transformer.py
View File

@ -40,7 +40,7 @@ class RotationTransformer:
getattr(pt, f'matrix_to_{from_rep}') getattr(pt, f'matrix_to_{from_rep}')
] ]
if from_convention is not None: if from_convention is not None:
funcs = [functools.partial(func, convernsion=from_convention) funcs = [functools.partial(func, convention=from_convention)
for func in funcs] for func in funcs]
forward_funcs.append(funcs[0]) forward_funcs.append(funcs[0])
inverse_funcs.append(funcs[1]) inverse_funcs.append(funcs[1])
@ -51,7 +51,7 @@ class RotationTransformer:
getattr(pt, f'{to_rep}_to_matrix') getattr(pt, f'{to_rep}_to_matrix')
] ]
if to_convention is not None: if to_convention is not None:
funcs = [functools.partial(func, convernsion=to_convention) funcs = [functools.partial(func, convention=to_convention)
for func in funcs] for func in funcs]
forward_funcs.append(funcs[0]) forward_funcs.append(funcs[0])
inverse_funcs.append(funcs[1]) inverse_funcs.append(funcs[1])