diff --git a/.gitignore b/.gitignore index 2798d1f..9369669 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,5 @@ __pycache__/ mini_bdx_runtime.egg-info/ build -dist \ No newline at end of file +dist +.DS_Store diff --git a/scripts/configure_all_motors.py b/dev/hardware/configure_all_motors.py similarity index 100% rename from scripts/configure_all_motors.py rename to dev/hardware/configure_all_motors.py diff --git a/scripts/head_puppet.py b/dev/hardware/head_puppet.py similarity index 100% rename from scripts/head_puppet.py rename to dev/hardware/head_puppet.py diff --git a/scripts/imu_client.py b/dev/hardware/imu_client.py similarity index 100% rename from scripts/imu_client.py rename to dev/hardware/imu_client.py diff --git a/scripts/imu_server.py b/dev/hardware/imu_server.py similarity index 100% rename from scripts/imu_server.py rename to dev/hardware/imu_server.py diff --git a/scripts/new_record_data.py b/dev/software/new_record_data.py similarity index 100% rename from scripts/new_record_data.py rename to dev/software/new_record_data.py diff --git a/scripts/plot_recorded_data.py b/dev/software/plot_recorded_data.py similarity index 100% rename from scripts/plot_recorded_data.py rename to dev/software/plot_recorded_data.py diff --git a/scripts/record_data.py b/dev/software/record_data.py similarity index 100% rename from scripts/record_data.py rename to dev/software/record_data.py diff --git a/mini_bdx_runtime/__init__.py b/mini_bdx_runtime/__init__.py index 8b13789..0d1570c 100644 --- a/mini_bdx_runtime/__init__.py +++ b/mini_bdx_runtime/__init__.py @@ -1 +1 @@ - +from .onnx_infer import OnnxInfer diff --git a/mini_bdx_runtime/mini_bdx_runtime/antennas.py b/mini_bdx_runtime/antennas.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/antennas.py rename to mini_bdx_runtime/antennas.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/buttons.py b/mini_bdx_runtime/buttons.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/buttons.py rename to mini_bdx_runtime/buttons.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/camera.py b/mini_bdx_runtime/camera.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/camera.py rename to mini_bdx_runtime/camera.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/duck_config.py b/mini_bdx_runtime/duck_config.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/duck_config.py rename to mini_bdx_runtime/duck_config.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/eyes.py b/mini_bdx_runtime/eyes.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/eyes.py rename to mini_bdx_runtime/eyes.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/feet_contacts.py b/mini_bdx_runtime/feet_contacts.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/feet_contacts.py rename to mini_bdx_runtime/feet_contacts.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/imu.py b/mini_bdx_runtime/imu.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/imu.py rename to mini_bdx_runtime/imu.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/__init__.py b/mini_bdx_runtime/mini_bdx_runtime/__init__.py deleted file mode 100644 index 0d1570c..0000000 --- a/mini_bdx_runtime/mini_bdx_runtime/__init__.py +++ /dev/null @@ -1 +0,0 @@ -from .onnx_infer import OnnxInfer diff --git a/mini_bdx_runtime/mini_bdx_runtime/onnx_infer.py b/mini_bdx_runtime/onnx_infer.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/onnx_infer.py rename to mini_bdx_runtime/onnx_infer.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/poly_reference_motion.py b/mini_bdx_runtime/poly_reference_motion.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/poly_reference_motion.py rename to mini_bdx_runtime/poly_reference_motion.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/projector.py b/mini_bdx_runtime/projector.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/projector.py rename to mini_bdx_runtime/projector.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/raw_imu.py b/mini_bdx_runtime/raw_imu.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/raw_imu.py rename to mini_bdx_runtime/raw_imu.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/rl_utils.py b/mini_bdx_runtime/rl_utils.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/rl_utils.py rename to mini_bdx_runtime/rl_utils.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/rustypot_position_hwi.py b/mini_bdx_runtime/rustypot_position_hwi.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/rustypot_position_hwi.py rename to mini_bdx_runtime/rustypot_position_hwi.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/sounds.py b/mini_bdx_runtime/sounds.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/sounds.py rename to mini_bdx_runtime/sounds.py diff --git a/mini_bdx_runtime/mini_bdx_runtime/xbox_controller.py b/mini_bdx_runtime/xbox_controller.py similarity index 100% rename from mini_bdx_runtime/mini_bdx_runtime/xbox_controller.py rename to mini_bdx_runtime/xbox_controller.py diff --git a/scripts/polynomial_coefficients.pkl b/polynomial_coefficients.pkl similarity index 100% rename from scripts/polynomial_coefficients.pkl rename to polynomial_coefficients.pkl diff --git a/scripts/v2_rl_walk_mujoco.py b/run_rl_walk.py similarity index 99% rename from scripts/v2_rl_walk_mujoco.py rename to run_rl_walk.py index 530ebb9..2e3977f 100644 --- a/scripts/v2_rl_walk_mujoco.py +++ b/run_rl_walk.py @@ -115,7 +115,7 @@ class RLWalk: self.projector = Projector() if self.duck_config.speaker: self.sounds = Sounds( - volume=1.0, sound_directory="../mini_bdx_runtime/assets/" + volume=1.0, sound_directory="src/mini_bdx_runtime/assets/" ) if self.duck_config.antennas: self.antennas = Antennas() diff --git a/scripts/antennas_controller_test.py b/tests/antennas_controller_test.py similarity index 100% rename from scripts/antennas_controller_test.py rename to tests/antennas_controller_test.py diff --git a/scripts/cam_test.py b/tests/cam_test.py similarity index 100% rename from scripts/cam_test.py rename to tests/cam_test.py diff --git a/scripts/fc_test.py b/tests/fc_test.py similarity index 100% rename from scripts/fc_test.py rename to tests/fc_test.py diff --git a/scripts/check_motors.py b/tests/test_motor_controller.py similarity index 100% rename from scripts/check_motors.py rename to tests/test_motor_controller.py diff --git a/scripts/calibrate_imu.py b/tools/calibrate_imu.py similarity index 100% rename from scripts/calibrate_imu.py rename to tools/calibrate_imu.py diff --git a/scripts/check_voltage.py b/tools/check_voltage.py similarity index 100% rename from scripts/check_voltage.py rename to tools/check_voltage.py diff --git a/scripts/configure_motor.py b/tools/configure_motor.py similarity index 100% rename from scripts/configure_motor.py rename to tools/configure_motor.py diff --git a/scripts/find_soft_offsets.py b/tools/find_soft_offsets.py similarity index 100% rename from scripts/find_soft_offsets.py rename to tools/find_soft_offsets.py diff --git a/scripts/turn_off.py b/tools/turn_off.py similarity index 100% rename from scripts/turn_off.py rename to tools/turn_off.py diff --git a/scripts/turn_on.py b/tools/turn_on.py similarity index 100% rename from scripts/turn_on.py rename to tools/turn_on.py