dexrtgt 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. dexrtgt-0.1.0.dist-info/METADATA +65 -0
  2. dexrtgt-0.1.0.dist-info/RECORD +103 -0
  3. dexrtgt-0.1.0.dist-info/WHEEL +5 -0
  4. dexrtgt-0.1.0.dist-info/licenses/LICENSE +21 -0
  5. dexrtgt-0.1.0.dist-info/licenses/LICENSE-THIRD-PARTY +32 -0
  6. dexrtgt-0.1.0.dist-info/top_level.txt +1 -0
  7. robokit/__init__.py +31 -0
  8. robokit/helpers/__init__.py +12 -0
  9. robokit/helpers/humanoid_retarget/__init__.py +0 -0
  10. robokit/helpers/humanoid_retarget/humanoid_retargeting_online.py +535 -0
  11. robokit/helpers/humanoid_retarget/motion_viewer.py +535 -0
  12. robokit/helpers/humanoid_retarget/utils/bvh_loader.py +487 -0
  13. robokit/helpers/humanoid_retarget/utils/smplx_loader.py +122 -0
  14. robokit/helpers/ik.py +921 -0
  15. robokit/helpers/ik_cpu.py +122 -0
  16. robokit/lie/__init__.py +5 -0
  17. robokit/lie/pinocchio_se3.py +143 -0
  18. robokit/lie/pinocchio_so3.py +103 -0
  19. robokit/lie/se3.py +155 -0
  20. robokit/lie/so3.py +147 -0
  21. robokit/lie/torch_se3.py +407 -0
  22. robokit/lie/torch_so3.py +315 -0
  23. robokit/lie/warp_se3.py +530 -0
  24. robokit/lie/warp_se3_kernels.py +1118 -0
  25. robokit/lie/warp_so3.py +333 -0
  26. robokit/lie/warp_so3_kernels.py +111 -0
  27. robokit/opt/__init__.py +5 -0
  28. robokit/opt/numpy_optimizer.py +134 -0
  29. robokit/opt/optimizer.py +48 -0
  30. robokit/opt/sparse_warp_optimizer.py +682 -0
  31. robokit/opt/torch_optimizer.py +111 -0
  32. robokit/opt/var_values.py +129 -0
  33. robokit/opt/variables.py +41 -0
  34. robokit/opt/warp_optimizer.py +493 -0
  35. robokit/opt/warp_solver.py +316 -0
  36. robokit/py.typed +0 -0
  37. robokit/robo/__init__.py +5 -0
  38. robokit/robo/numpy_robot.py +427 -0
  39. robokit/robo/robot.py +208 -0
  40. robokit/robo/robot_spec.py +754 -0
  41. robokit/robo/torch_robot.py +549 -0
  42. robokit/robo/warp_robot.py +791 -0
  43. robokit/robo/warp_robot_kernels.py +819 -0
  44. robokit/robo/yourdfpy.py +31 -0
  45. robokit/terms/__init__.py +73 -0
  46. robokit/terms/numpy/__init__.py +0 -0
  47. robokit/terms/numpy/base_damping_task.py +88 -0
  48. robokit/terms/numpy/base_step_limit.py +56 -0
  49. robokit/terms/numpy/frame_task.py +103 -0
  50. robokit/terms/numpy/position_limit.py +77 -0
  51. robokit/terms/numpy/rest_task.py +105 -0
  52. robokit/terms/numpy/smoothness_task.py +129 -0
  53. robokit/terms/numpy/velocity_limit_task.py +138 -0
  54. robokit/terms/terms.py +314 -0
  55. robokit/terms/torch/__init__.py +0 -0
  56. robokit/terms/torch/base_damping_task.py +88 -0
  57. robokit/terms/torch/base_step_limit.py +72 -0
  58. robokit/terms/torch/frame_task.py +72 -0
  59. robokit/terms/torch/position_limit.py +68 -0
  60. robokit/terms/torch/rest_task.py +115 -0
  61. robokit/terms/torch/smoothness_task.py +137 -0
  62. robokit/terms/torch/velocity_limit_task.py +112 -0
  63. robokit/terms/warp/__init__.py +0 -0
  64. robokit/terms/warp/base_damping_task.py +274 -0
  65. robokit/terms/warp/base_step_limit.py +235 -0
  66. robokit/terms/warp/collision_task.py +533 -0
  67. robokit/terms/warp/contact_attraction_task.py +263 -0
  68. robokit/terms/warp/distance_task.py +896 -0
  69. robokit/terms/warp/frame_position_huber_task.py +308 -0
  70. robokit/terms/warp/frame_retargeting_task.py +419 -0
  71. robokit/terms/warp/frame_task.py +418 -0
  72. robokit/terms/warp/frame_vector_distance_task.py +380 -0
  73. robokit/terms/warp/position_limit.py +187 -0
  74. robokit/terms/warp/position_score.py +173 -0
  75. robokit/terms/warp/position_task.py +228 -0
  76. robokit/terms/warp/rest_task.py +320 -0
  77. robokit/terms/warp/rotation_task.py +226 -0
  78. robokit/terms/warp/self_penetration_task.py +312 -0
  79. robokit/terms/warp/smoothness_task.py +366 -0
  80. robokit/terms/warp/velocity_limit_task.py +242 -0
  81. robokit/thirdparty/__init__.py +0 -0
  82. robokit/thirdparty/pytorch3d/__init__.py +0 -0
  83. robokit/thirdparty/pytorch3d/math.py +87 -0
  84. robokit/thirdparty/pytorch3d/rotation_conversions.py +646 -0
  85. robokit/thirdparty/pytorch3d/se3.py +223 -0
  86. robokit/thirdparty/pytorch3d/so3.py +269 -0
  87. robokit/types.py +16 -0
  88. robokit/utils/__init__.py +0 -0
  89. robokit/utils/tensor_utils.py +264 -0
  90. robokit/utils/visualize_utils.py +95 -0
  91. robokit/utils/warp_utils.py +444 -0
  92. robokit/xform/__init__.py +0 -0
  93. robokit/xform/numpy/__init__.py +32 -0
  94. robokit/xform/numpy/rotation_conversions.py +363 -0
  95. robokit/xform/numpy/transforms.py +87 -0
  96. robokit/xform/torch/__init__.py +47 -0
  97. robokit/xform/torch/rotation_conversions.py +317 -0
  98. robokit/xform/torch/transforms.py +262 -0
  99. robokit/xform/torch/utils.py +298 -0
  100. robokit/xform/warp/__init__.py +0 -0
  101. robokit/xform/warp/rotation_conversions.py +261 -0
  102. robokit/xform/warp/torch_wrappers.py +992 -0
  103. robokit/xform/warp/transforms.py +46 -0
@@ -0,0 +1,65 @@
1
+ Metadata-Version: 2.4
2
+ Name: dexrtgt
3
+ Version: 0.1.0
4
+ Summary: dexrtgt
5
+ Author: dexrtgt
6
+ License: MIT
7
+ Classifier: Programming Language :: Python :: 3
8
+ Classifier: Programming Language :: Python :: 3.8
9
+ Classifier: Programming Language :: Python :: 3.9
10
+ Classifier: Programming Language :: Python :: 3.10
11
+ Classifier: Programming Language :: Python :: 3.11
12
+ Classifier: Programming Language :: Python :: 3.12
13
+ Classifier: License :: OSI Approved :: MIT License
14
+ Classifier: Operating System :: OS Independent
15
+ Requires-Python: <3.13,>=3.8
16
+ Description-Content-Type: text/markdown
17
+ License-File: LICENSE
18
+ License-File: LICENSE-THIRD-PARTY
19
+ Requires-Dist: packaging
20
+ Requires-Dist: typing-extensions
21
+ Requires-Dist: jaxtyping
22
+ Requires-Dist: rich
23
+ Requires-Dist: PyYAML
24
+ Requires-Dist: numpy<2
25
+ Requires-Dist: yourdfpy
26
+ Requires-Dist: pin
27
+ Requires-Dist: qpsolvers[daqp]
28
+ Requires-Dist: viser
29
+ Requires-Dist: einops
30
+ Requires-Dist: warp-lang
31
+ Requires-Dist: beartype>=0.19.0
32
+ Requires-Dist: smplx>=0.1.28
33
+ Requires-Dist: natsort>=8.4.0
34
+ Requires-Dist: robot-descriptions>=1.22.0
35
+ Provides-Extra: cpu
36
+ Requires-Dist: torch; extra == "cpu"
37
+ Provides-Extra: cu126
38
+ Requires-Dist: torch; extra == "cu126"
39
+ Provides-Extra: cu128
40
+ Requires-Dist: torch; extra == "cu128"
41
+ Requires-Dist: nvidia-nccl-cu12; extra == "cu128"
42
+ Provides-Extra: dev
43
+ Requires-Dist: robot_descriptions; extra == "dev"
44
+ Requires-Dist: pandas; extra == "dev"
45
+ Requires-Dist: tabulate; extra == "dev"
46
+ Requires-Dist: pin-pink; extra == "dev"
47
+ Requires-Dist: smplx; extra == "dev"
48
+ Requires-Dist: opencv-python; extra == "dev"
49
+ Requires-Dist: ipython; extra == "dev"
50
+ Requires-Dist: ruff; extra == "dev"
51
+ Requires-Dist: pyright; extra == "dev"
52
+ Requires-Dist: pytest; extra == "dev"
53
+ Requires-Dist: xdoctest; extra == "dev"
54
+ Requires-Dist: coverage; extra == "dev"
55
+ Requires-Dist: pytest-cov; extra == "dev"
56
+ Requires-Dist: pre-commit; extra == "dev"
57
+ Requires-Dist: tyro; extra == "dev"
58
+ Requires-Dist: gdown; extra == "dev"
59
+ Requires-Dist: tabulate>=0.9.0; extra == "dev"
60
+ Requires-Dist: ipdb>=0.13.13; extra == "dev"
61
+ Dynamic: license-file
62
+
63
+ # dexrtgt
64
+
65
+ GPU-accelerated inverse kinematics and humanoid motion retargeting.
@@ -0,0 +1,103 @@
1
+ dexrtgt-0.1.0.dist-info/licenses/LICENSE,sha256=HNUcyRQHZfg9FjR8MT1s7K_EIaSRxZr-MRFb7bXOu8s,1064
2
+ dexrtgt-0.1.0.dist-info/licenses/LICENSE-THIRD-PARTY,sha256=_eO5pQo8IwZIct-Q-M4KlYUYnlR4EO5XTEgPWb_apdY,1602
3
+ robokit/__init__.py,sha256=jBuMFz4Bcp-VSNG0zqznR5CvuwNJunmKro6lH1CCRlc,707
4
+ robokit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
+ robokit/types.py,sha256=5m9ZKAX3km1rZIqenl44LiPzKOoUn-wHTp94G9EuSd8,255
6
+ robokit/helpers/__init__.py,sha256=ymxYo4IwV5Ypza6SpFj9k8jqYanpWq3jyYC4PuhSdhQ,295
7
+ robokit/helpers/ik.py,sha256=SBxW_LlYR7Y7VD6F7EopaUKZdL_MB8kyKmbzj3O5D7w,41318
8
+ robokit/helpers/ik_cpu.py,sha256=tNRwmMyhUqK75bxv7zLQDzaOqH_ctbA1HQ1x9pwuBgo,4611
9
+ robokit/helpers/humanoid_retarget/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
+ robokit/helpers/humanoid_retarget/humanoid_retargeting_online.py,sha256=M4mSNMFJBfFBvoAiRGCrfgyeTvcTYelcVUzlEhiz0YA,22252
11
+ robokit/helpers/humanoid_retarget/motion_viewer.py,sha256=9H37YXIzmiEKCSBxbcJ2wtxWsHQ8XF76Ym7atkjMD1w,18445
12
+ robokit/helpers/humanoid_retarget/utils/bvh_loader.py,sha256=9bUk9z_scwLVXKZOjsAWPygj-_hvmEUI5omYWLpIeKM,17240
13
+ robokit/helpers/humanoid_retarget/utils/smplx_loader.py,sha256=IFbyTgfbUsRf-1GSoPFUTVUpGo5wc2N6vgOyi0wDa7g,4420
14
+ robokit/lie/__init__.py,sha256=XOImSnHlH5SKwCnqsYpuHBDm3V6Xa3ngVARl87YshlU,91
15
+ robokit/lie/pinocchio_se3.py,sha256=CF1huuGCkUO2Cs_WOvUolTwWJLfYIOFv7C8w3Mhnzg8,5009
16
+ robokit/lie/pinocchio_so3.py,sha256=rwA1m7r9a67BeK-sBbPTD0X1WHkmXi7Xb8vhmk6DpCw,3489
17
+ robokit/lie/se3.py,sha256=MTZTNNvCQwYU6Jck1JZKJMnijeCbYaFo6b9FxwcmaJs,5074
18
+ robokit/lie/so3.py,sha256=4voztKoylcBI5CJ-tPNVqCB0iHVo0iEB6TXVXN0DCrQ,4794
19
+ robokit/lie/torch_se3.py,sha256=v6AcJ9GcRwpBBhAAfZXbY05Ttn3yp4IoTQf2t-DPIL8,14912
20
+ robokit/lie/torch_so3.py,sha256=vagIwWtexp3XFyoykksTO_cZ9TjB7-I5IJIAuj-s_hY,10484
21
+ robokit/lie/warp_se3.py,sha256=jbBoL5WS5odEzjzmBrRUNXDw-FkMsRdE9W8rZZ0Vs_E,19311
22
+ robokit/lie/warp_se3_kernels.py,sha256=Sg1Dz-jfoXCB-IcFm-1a25GwS-dliGBTa0iJhYJL9sE,36861
23
+ robokit/lie/warp_so3.py,sha256=IVChI0ownJj5fsxQXopPk8BZo0YT29JEDtlo_9UrM-M,11715
24
+ robokit/lie/warp_so3_kernels.py,sha256=ytVn05RGcDVbDlNDkNciulYXa5zTBAFZQchOzf2aY1Y,3291
25
+ robokit/opt/__init__.py,sha256=vObp4gUFVsaYjJa-HuHjAz5vCPMann98xaFFFplgZ40,144
26
+ robokit/opt/numpy_optimizer.py,sha256=JZRKjKhHzfmQ2lBy5mUwWYQeK8ZwBrwxRoJgJxyYyh4,5047
27
+ robokit/opt/optimizer.py,sha256=eX8jJ2ylsyOt46WzRKkkqd2kpO2f0LR1NY7_piBiRuE,1195
28
+ robokit/opt/sparse_warp_optimizer.py,sha256=DOAsuh3Q09YdHA2HvFC8Ade4WUY-F75ltrd-qwute1Q,25010
29
+ robokit/opt/torch_optimizer.py,sha256=kny5ZYMIRw8bHBdeaXsCk03aGu13UyFSjXW2-msuGQM,4410
30
+ robokit/opt/var_values.py,sha256=170mNaiVTf3hYYlEqLCT2dxO2CyTUjvoznb30x_seXc,4486
31
+ robokit/opt/variables.py,sha256=eQEJfpQHUpxUasZjezqU7TC565-KqEzq07_x9pl-oj0,831
32
+ robokit/opt/warp_optimizer.py,sha256=bqUZC8NunGvAlRPjcTALxKmakAqpnkkCv3j_cVA8-vg,19899
33
+ robokit/opt/warp_solver.py,sha256=0QRm_rpo-RtA-RIkrEu33Aat5rbwd7VylXHuDyGXhnE,13079
34
+ robokit/robo/__init__.py,sha256=bmVQq1rhQtAgSl7jAP2OxOcN8lz_F8RAuAR7-5Vq0sc,118
35
+ robokit/robo/numpy_robot.py,sha256=kv6c83Bw3P3WBQzZW7VIAJMQWrwF2WDc2ewd8VjS9LM,20965
36
+ robokit/robo/robot.py,sha256=ztA23dg_XDn-j-0cHDJk2LwukOhDThE8VXjoiRjDy_c,7805
37
+ robokit/robo/robot_spec.py,sha256=szGE4zcw5BFn0g6gWrHGChNIp_bFmRq2pk6ur1vhOMs,34169
38
+ robokit/robo/torch_robot.py,sha256=H69Nlb2wgBDB1YgqcmsGdg2nR7HNDyCOUw0KSvXkjjU,27161
39
+ robokit/robo/warp_robot.py,sha256=rBSeAJspqWLDVHEag0vooLRC2_KvGlx40I38WyRL3CY,34404
40
+ robokit/robo/warp_robot_kernels.py,sha256=QpZi7_g8T_UyQDolG0BlrEL4n0XS5A4klXU870NwEJU,31571
41
+ robokit/robo/yourdfpy.py,sha256=rvRwm1LFEaJ-iKlvZuHO_DNbxy1vvq95lSPUBcNsfdE,1024
42
+ robokit/terms/__init__.py,sha256=WhkDK5noSU0cGgaIz2qcqmzjRhHMYfs5FZ_INvoEmw8,3126
43
+ robokit/terms/terms.py,sha256=2iQriU0WZAq6B7z38SeKvvc7vtBKr3xRgA3c7Gk8qDA,11766
44
+ robokit/terms/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
45
+ robokit/terms/numpy/base_damping_task.py,sha256=-0I0O-NBUGFOHCWUr0y7B_I-h-dPijTDB1jLT2DVJRs,3121
46
+ robokit/terms/numpy/base_step_limit.py,sha256=YtnPIEVPPEpcSLZKk83QhapKSoXk1JKe953Dj28F2j8,2224
47
+ robokit/terms/numpy/frame_task.py,sha256=vR8LBZTcTVkjOB1xju4YBOxGDnIGzL1HtuoBrTJvl3o,4362
48
+ robokit/terms/numpy/position_limit.py,sha256=nWVj9oVZghrrbGYkgK3RJPqLExGWkMhASQgFnoCDMEo,3343
49
+ robokit/terms/numpy/rest_task.py,sha256=9G8HuIkiQ-oW9tlourEDiv7Kmiattz0a46KbJ7SpAwM,4136
50
+ robokit/terms/numpy/smoothness_task.py,sha256=floNAgpRyLHGUoF214rHOetJa9jew2EDtzUyb51AkBM,5173
51
+ robokit/terms/numpy/velocity_limit_task.py,sha256=aDH1I8H57kb4Wk8D6MR7KO5aETaRhofXne59l3WGVLY,5380
52
+ robokit/terms/torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
53
+ robokit/terms/torch/base_damping_task.py,sha256=U2XedbcmRO4dLO5QOMOieHV79k9jW5VJQ5o3u7f_VjI,3121
54
+ robokit/terms/torch/base_step_limit.py,sha256=QRtf3DJpEKkdk24IzobTJDdgRA9C3LYKzE1we1IghSQ,2796
55
+ robokit/terms/torch/frame_task.py,sha256=rY1r-C6cHZs8MDBTeVDbYFXZ9NV7otORKMExUFbwU0I,3085
56
+ robokit/terms/torch/position_limit.py,sha256=GHqcbqhvB7DIZ9YL697VxiD8Hbgg1aI8XeMoacia7fg,2811
57
+ robokit/terms/torch/rest_task.py,sha256=zI7Mr_P0FwbY4UEONVlqxbUpyoUJcya82oHt5IpkPDU,4457
58
+ robokit/terms/torch/smoothness_task.py,sha256=XNXF0ES4yZTXOmqpOUAYhor-uxxyXoA0vvx9AfSXBtE,5493
59
+ robokit/terms/torch/velocity_limit_task.py,sha256=u_mX8dNK9-6JXf44B8iK15l7rHJAzwn1eX3q7dtTsJM,4253
60
+ robokit/terms/warp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
61
+ robokit/terms/warp/base_damping_task.py,sha256=A2d8lIiCaBfu_zJIPtlCFVxBZkf2d6VyGZjjuS7yUh0,11031
62
+ robokit/terms/warp/base_step_limit.py,sha256=J7Bf5TU77x_tteLudznovLOIJ7qyye1docmsqy928jg,9512
63
+ robokit/terms/warp/collision_task.py,sha256=zYrSr2XI6zVOfHhCt9IqU14W9bufoMuotNL2J8GxR0o,21256
64
+ robokit/terms/warp/contact_attraction_task.py,sha256=SdziLuICFS7jUd_O-GWU7sjEQ50VElPTgwdWdTQ1ifY,9569
65
+ robokit/terms/warp/distance_task.py,sha256=G05UMbFfy3AG2LHc1nKhkFObGXMZ1Rdy-6jqCAXAQtU,37452
66
+ robokit/terms/warp/frame_position_huber_task.py,sha256=1HcrqOwCcnT9UOPvvks_XBKYzLXL6UQYHkM8u_WYrQc,11279
67
+ robokit/terms/warp/frame_retargeting_task.py,sha256=iKBCQiK58yc8SAmjk-5JlIeNaiQ_hirnqZKEMy9Fh2Y,15959
68
+ robokit/terms/warp/frame_task.py,sha256=r6z3kmGQRWSQLUpn71K_0QTqdzgB0K3szJqZxW0SQKs,17271
69
+ robokit/terms/warp/frame_vector_distance_task.py,sha256=t8vrmMIlW9nQK0LCR5EP1K2OU4sFb-aLIJJQnJz_OHQ,14252
70
+ robokit/terms/warp/position_limit.py,sha256=ynG_DMj0rh7OPpQgbE_XJPl3YSmHxW2qgjHPjFijxyo,6902
71
+ robokit/terms/warp/position_score.py,sha256=WDcJb4IU7T5iwhOQhCS3SVbKznbe9imI8Dq_Mk9YPpU,5800
72
+ robokit/terms/warp/position_task.py,sha256=meBG6oV8ifOeBxD5AXonlx7hNuZbPkxfiqhdxVT2aiM,9579
73
+ robokit/terms/warp/rest_task.py,sha256=-oF1pMK2C_RwBInn08EhjPLq4qm13IK2Lv0jeJaH55Y,13089
74
+ robokit/terms/warp/rotation_task.py,sha256=W69fNINhZkA4lqMvIUUx18WT07KsC1YKFs3AEa-uO7E,9412
75
+ robokit/terms/warp/self_penetration_task.py,sha256=b8YEYvkluwvJKp9djKtmo5vAsBJjzaFuWWpwk1lhZbA,11353
76
+ robokit/terms/warp/smoothness_task.py,sha256=4UBYUd5pW--1NW68Ap-Ary2kpJ5E3yMjaXDnM15jRqM,15021
77
+ robokit/terms/warp/velocity_limit_task.py,sha256=P29P_aTOI79UUDg-uqFvxUvGW_rCCP01Q-LZg9AHsg4,9142
78
+ robokit/thirdparty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
79
+ robokit/thirdparty/pytorch3d/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
80
+ robokit/thirdparty/pytorch3d/math.py,sha256=wZwo5j3KLYVhHDA_bxD77OYnQnjm2QDKRg3hSfp6enw,2981
81
+ robokit/thirdparty/pytorch3d/rotation_conversions.py,sha256=uc6GRH3FVjMUV57eX4D9EREMuEpm6T_cuQCD4VJAKmc,22448
82
+ robokit/thirdparty/pytorch3d/se3.py,sha256=nNLkIlittgCNUv1gNVy5KLMek2F7n9XA6iuHghpbbVI,8028
83
+ robokit/thirdparty/pytorch3d/so3.py,sha256=cA9NahF0U6AbpiOUMXMX2tYd3fRUyxjewrPqT-rRA28,8685
84
+ robokit/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
85
+ robokit/utils/tensor_utils.py,sha256=OQhYo5IKLy6J_qh7hIv0VtIAE_sZYW-qL7gYE8AZDmg,10148
86
+ robokit/utils/visualize_utils.py,sha256=owoOqOqgSUw8kPyq8O0vSWflvEi7F5SQCYZM3FokB5w,3355
87
+ robokit/utils/warp_utils.py,sha256=yZjxDNUDbFlGklnJXkzi5WUuFXNbyO4ezpfwQ_kiYh4,12028
88
+ robokit/xform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
89
+ robokit/xform/numpy/__init__.py,sha256=gwOsjarUgWDo8N2Qy8pCJArYYgV2sXIhv1RX2bWeseg,811
90
+ robokit/xform/numpy/rotation_conversions.py,sha256=dg5vpUUlMBbytyVLUZFKVCo_7arQDZKpvw51vuJThFE,12490
91
+ robokit/xform/numpy/transforms.py,sha256=Sp3GMctq0eBuBT4lGNifTIMG4uXt78mngp6tvABwouQ,3155
92
+ robokit/xform/torch/__init__.py,sha256=TlciziR7VNz0ZWjCr3y9YTtpRoqNynSIXGLrX92u70U,1237
93
+ robokit/xform/torch/rotation_conversions.py,sha256=NNEtcBTCrXmoYKvk8SnlQL2MvuelnNdJXn2w_wIu3gk,11758
94
+ robokit/xform/torch/transforms.py,sha256=O6UPM72bbs8q-GP4gAf6f6sz6B1UTlVun2tAW6vaPc4,9913
95
+ robokit/xform/torch/utils.py,sha256=VawAjGjBwKSUoA0wRQl2RGdk6qzP5GXVVUo2uLl5zdA,11367
96
+ robokit/xform/warp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
97
+ robokit/xform/warp/rotation_conversions.py,sha256=qOFEcRuJKol7sL0rO25MtnyF6eyvSQlNteTqb7Pv6kI,8279
98
+ robokit/xform/warp/torch_wrappers.py,sha256=P3A_lP8vzWk1ynJbImf-Ud5A5CjW9eSdFVer_Fuqy68,37161
99
+ robokit/xform/warp/transforms.py,sha256=gYhUkJ5Lnkn8_wHQgeFmoDZoWn0qDz_BAGkPRoppeXc,1142
100
+ dexrtgt-0.1.0.dist-info/METADATA,sha256=egZS-_hLiQcI0So8BVqHq3PEjRcGAvZHcCPG0Dy7F3g,2126
101
+ dexrtgt-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
102
+ dexrtgt-0.1.0.dist-info/top_level.txt,sha256=W6wLzL_Jf3fG1cAJJDvX8uwPApBq4tJkMOczS_-Bj_o,8
103
+ dexrtgt-0.1.0.dist-info/RECORD,,
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (82.0.1)
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
5
+
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2025 dexrtgt
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,32 @@
1
+ PyTorch3D License (https://github.com/facebookresearch/pytorch3d):
2
+
3
+ BSD License
4
+
5
+ For PyTorch3D software
6
+
7
+ Copyright (c) Meta Platforms, Inc. and affiliates. All rights reserved.
8
+
9
+ Redistribution and use in source and binary forms, with or without modification,
10
+ are permitted provided that the following conditions are met:
11
+
12
+ * Redistributions of source code must retain the above copyright notice, this
13
+ list of conditions and the following disclaimer.
14
+
15
+ * Redistributions in binary form must reproduce the above copyright notice,
16
+ this list of conditions and the following disclaimer in the documentation
17
+ and/or other materials provided with the distribution.
18
+
19
+ * Neither the name Meta nor the names of its contributors may be used to
20
+ endorse or promote products derived from this software without specific
21
+ prior written permission.
22
+
23
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24
+ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25
+ WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
26
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
27
+ ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
28
+ (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
29
+ LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
30
+ ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
32
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
@@ -0,0 +1 @@
1
+ robokit
robokit/__init__.py ADDED
@@ -0,0 +1,31 @@
1
+ import logging
2
+ from dataclasses import dataclass
3
+
4
+ import rich
5
+ from rich.logging import RichHandler
6
+
7
+
8
+ # ----------------------------- Logging -----------------------------
9
+ rich.reconfigure(log_path=False)
10
+
11
+ logger = logging.getLogger("robokit")
12
+ logger.propagate = False
13
+ logger.setLevel("INFO")
14
+ if not logger.handlers:
15
+ logger.addHandler(RichHandler(console=rich.get_console(), show_path=False, log_time_format="[%X]"))
16
+
17
+
18
+ # ----------------------------- Config -----------------------------
19
+ @dataclass
20
+ class Config:
21
+ """Configuration for robokit."""
22
+
23
+ enable_torch_jit: bool = True
24
+ """Whether to enable torch jit."""
25
+
26
+
27
+ CONFIG = Config()
28
+
29
+ __version__ = "0.1.0"
30
+
31
+ __all__ = ["CONFIG", "__version__"]
@@ -0,0 +1,12 @@
1
+ from robokit.helpers.ik import IKHelper, IKHelperConfig
2
+ from robokit.helpers.ik_cpu import CPUIKHelper, CPUIKHelperConfig
3
+ from robokit.opt.warp_solver import WarpStageConfig
4
+
5
+
6
+ __all__ = [
7
+ "CPUIKHelper",
8
+ "CPUIKHelperConfig",
9
+ "IKHelper",
10
+ "IKHelperConfig",
11
+ "WarpStageConfig",
12
+ ]
File without changes