added locomotion_rot

nagyrajmund · Dec 12, 2020 · da54567 · da54567
1 parent 0191b8d
commit da54567
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/motion/__init__.py b/motion/__init__.py
@@ -1,6 +1,7 @@
-from .datasets import Locomotion, Trinity
+from .datasets import Locomotion, Locomotion_rot, Trinity
 
 Datasets = {
     "locomotion": Locomotion,
+    "locomotion_rot": Locomotion_rot,
     "trinity": Trinity
 }