Fix python wheels

setup.py

@@ -17,5 +17,6 @@
             "cyclonedds==0.10.2",
             "numpy",
             "opencv-python",
+            "pygame",
       ],
-      )
+      )

unitree_sdk2py/__init__.py

@@ -1,10 +1,12 @@
-from . import idl, utils, core, rpc, go2, b2
+from . import idl, utils, core, rpc, go2, b2, g1, h1
 
 __all__ = [
-    "idl"
-    "utils"
+    "idl",
+    "utils",
     "core",
     "rpc",
     "go2",
     "b2",
+    "g1",
+    "h1",
 ]

unitree_sdk2py/b2/__init__.py

@@ -0,0 +1,13 @@
+from . import back_video
+from . import front_video
+from . import robot_state
+from . import sport
+from . import vui
+
+__all__ = [
+    "back_video",
+    "front_video",
+    "robot_state",
+    "sport",
+    "vui",
+]

unitree_sdk2py/b2/back_video/__init__.py

@@ -0,0 +1,7 @@
+from . import back_video_api
+from . import back_video_client
+
+__all__ = [
+    "back_video_api",
+    "back_video_client",
+]

unitree_sdk2py/b2/front_video/__init__.py

@@ -0,0 +1,7 @@
+from . import front_video_api
+from . import front_video_client
+
+__all__ = [
+    "front_video_api",
+    "front_video_client",
+]

unitree_sdk2py/b2/robot_state/__init__.py

@@ -0,0 +1,7 @@
+from . import robot_state_api
+from . import robot_state_client
+
+__all__ = [
+    "robot_state_api",
+    "robot_state_client",
+]

unitree_sdk2py/b2/robot_state/robot_state_client.py

@@ -1,7 +1,6 @@
 import json
 
 from ...rpc.client import Client
-from ...rpc.client_internal import *
 from .robot_state_api import *
 
 

unitree_sdk2py/b2/sport/__init__.py

@@ -0,0 +1,7 @@
+from . import sport_api
+from . import sport_client
+
+__all__ = [
+    "sport_api",
+    "sport_client",
+]

unitree_sdk2py/b2/vui/__init__.py

@@ -0,0 +1,7 @@
+from . import vui_api
+from . import vui_client
+
+__all__ = [
+    "vui_api",
+    "vui_client",
+]

unitree_sdk2py/comm/__init__.py

@@ -0,0 +1,5 @@
+from . import motion_switcher
+
+__all__ = [
+    "motion_switcher",
+]

unitree_sdk2py/comm/motion_switcher/__init__.py

@@ -0,0 +1,7 @@
+from . import motion_switcher_api
+from . import motion_switcher_client
+
+__all__ = [
+    "motion_switcher_api",
+    "motion_switcher_client",
+]

unitree_sdk2py/core/__init__.py

@@ -0,0 +1,9 @@
+from . import channel_config
+from . import channel_name
+from . import channel
+
+__all__ = [
+    "channel_config",
+    "channel_name",
+    "channel",
+]

unitree_sdk2py/g1/__init__.py

@@ -0,0 +1,7 @@
+from . import audio
+from . import loco
+
+__all__ = [
+    "audio",
+    "loco",
+]

unitree_sdk2py/g1/audio/__init__.py

@@ -0,0 +1,7 @@
+from . import g1_audio_api
+from . import g1_audio_client
+
+__all__ = [
+    "g1_audio_api",
+    "g1_audio_client",
+]

unitree_sdk2py/g1/loco/__init__.py

@@ -0,0 +1,7 @@
+from . import g1_loco_api
+from . import g1_loco_client
+
+__all__ = [
+    "g1_loco_api",
+    "g1_loco_client",
+]

unitree_sdk2py/go2/__init__.py

@@ -0,0 +1,13 @@
+from . import obstacles_avoid
+from . import robot_state
+from . import sport
+from . import video
+from . import vui
+
+__all__ = [
+    "obstacles_avoid",
+    "robot_state",
+    "sport",
+    "video",
+    "vui",
+]

unitree_sdk2py/go2/obstacles_avoid/__init__.py

@@ -0,0 +1,7 @@
+from . import obstacles_avoid_api
+from . import obstacles_avoid_client
+
+__all__ = [
+    "obstacles_avoid_api",
+    "obstacles_avoid_client",
+]

unitree_sdk2py/go2/robot_state/__init__.py

@@ -0,0 +1,7 @@
+from . import robot_state_api
+from . import robot_state_client
+
+__all__ = [
+    "robot_state_api",
+    "robot_state_client",
+]

unitree_sdk2py/go2/sport/__init__.py

@@ -0,0 +1,7 @@
+from . import sport_api
+from . import sport_client
+
+__all__ = [
+    "sport_api",
+    "sport_client",
+]

unitree_sdk2py/go2/video/__init__.py

@@ -0,0 +1,7 @@
+from . import video_api
+from . import video_client
+
+__all__ = [
+    "video_api",
+    "video_client",
+]

unitree_sdk2py/go2/vui/__init__.py

@@ -0,0 +1,7 @@
+from . import vui_api
+from . import vui_client
+
+__all__ = [
+    "vui_api",
+    "vui_client",
+]

unitree_sdk2py/h1/__init__.py

@@ -0,0 +1,5 @@
+from . import loco
+
+__all__ = [
+    "loco",
+]

unitree_sdk2py/h1/loco/__init__.py

@@ -0,0 +1,7 @@
+from . import h1_loco_api
+from . import h1_loco_client
+
+__all__ = [
+    "h1_loco_api",
+    "h1_loco_client",
+]

unitree_sdk2py/idl/__init__.py

@@ -1,5 +1,5 @@
 from .default import *
-from . import builtin_interfaces, geometry_msgs, sensor_msgs, std_msgs, unitree_go, unitree_api
+from . import builtin_interfaces, geometry_msgs, sensor_msgs, std_msgs, unitree_go, unitree_api, unitree_hg
 
 __all__ = [
     "builtin_interfaces",

unitree_sdk2py/rpc/__init__.py

@@ -0,0 +1,24 @@
+from . import client_base
+from . import client_stub
+from . import client
+from . import internal
+from . import lease_client
+from . import lease_server
+from . import request_future
+from . import server_base
+from . import server_stub
+from . import server
+
+
+__all__ = [
+    "client_base",
+    "client_stub",
+    "client",
+    "lease_client",
+    "lease_server",
+    "request_future",
+    "server_base",
+    "server_stub",
+    "server",
+    "internal",
+]