Merge pull request #27 from pointW/main

Fix typo in rotation_transformer.py
real-stanford · Oct 27, 2023 · 548a52b · 548a52b
2 parents 7dd9dc4 + de4384e
commit 548a52b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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])