chore(utils): move encoding utils and process to their respective modules (#2029)
Signed-off-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
@@ -22,7 +22,7 @@ import logging
|
|||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
|
||||||
from lerobot.utils.encoding_utils import decode_twos_complement, encode_twos_complement
|
from lerobot.motors.encoding_utils import decode_twos_complement, encode_twos_complement
|
||||||
|
|
||||||
from ..motors_bus import Motor, MotorCalibration, MotorsBus, NameOrID, Value, get_address
|
from ..motors_bus import Motor, MotorCalibration, MotorsBus, NameOrID, Value, get_address
|
||||||
from .tables import (
|
from .tables import (
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ from copy import deepcopy
|
|||||||
from enum import Enum
|
from enum import Enum
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
|
|
||||||
from lerobot.utils.encoding_utils import decode_sign_magnitude, encode_sign_magnitude
|
from lerobot.motors.encoding_utils import decode_sign_magnitude, encode_sign_magnitude
|
||||||
|
|
||||||
from ..motors_bus import Motor, MotorCalibration, MotorsBus, NameOrID, Value, get_address
|
from ..motors_bus import Motor, MotorCalibration, MotorsBus, NameOrID, Value, get_address
|
||||||
from .tables import (
|
from .tables import (
|
||||||
|
|||||||
@@ -63,6 +63,7 @@ from lerobot.configs.train import TrainRLServerPipelineConfig
|
|||||||
from lerobot.policies.factory import make_policy
|
from lerobot.policies.factory import make_policy
|
||||||
from lerobot.policies.sac.modeling_sac import SACPolicy
|
from lerobot.policies.sac.modeling_sac import SACPolicy
|
||||||
from lerobot.processor import TransitionKey
|
from lerobot.processor import TransitionKey
|
||||||
|
from lerobot.rl.process import ProcessSignalHandler
|
||||||
from lerobot.robots import so100_follower # noqa: F401
|
from lerobot.robots import so100_follower # noqa: F401
|
||||||
from lerobot.teleoperators import gamepad, so101_leader # noqa: F401
|
from lerobot.teleoperators import gamepad, so101_leader # noqa: F401
|
||||||
from lerobot.teleoperators.utils import TeleopEvents
|
from lerobot.teleoperators.utils import TeleopEvents
|
||||||
@@ -75,7 +76,6 @@ from lerobot.transport.utils import (
|
|||||||
send_bytes_in_chunks,
|
send_bytes_in_chunks,
|
||||||
transitions_to_bytes,
|
transitions_to_bytes,
|
||||||
)
|
)
|
||||||
from lerobot.utils.process import ProcessSignalHandler
|
|
||||||
from lerobot.utils.queue import get_last_item_from_queue
|
from lerobot.utils.queue import get_last_item_from_queue
|
||||||
from lerobot.utils.random_utils import set_seed
|
from lerobot.utils.random_utils import set_seed
|
||||||
from lerobot.utils.robot_utils import busy_wait
|
from lerobot.utils.robot_utils import busy_wait
|
||||||
|
|||||||
@@ -67,6 +67,7 @@ from lerobot.datasets.lerobot_dataset import LeRobotDataset
|
|||||||
from lerobot.policies.factory import make_policy
|
from lerobot.policies.factory import make_policy
|
||||||
from lerobot.policies.sac.modeling_sac import SACPolicy
|
from lerobot.policies.sac.modeling_sac import SACPolicy
|
||||||
from lerobot.rl.buffer import ReplayBuffer, concatenate_batch_transitions
|
from lerobot.rl.buffer import ReplayBuffer, concatenate_batch_transitions
|
||||||
|
from lerobot.rl.process import ProcessSignalHandler
|
||||||
from lerobot.robots import so100_follower # noqa: F401
|
from lerobot.robots import so100_follower # noqa: F401
|
||||||
from lerobot.teleoperators import gamepad, so101_leader # noqa: F401
|
from lerobot.teleoperators import gamepad, so101_leader # noqa: F401
|
||||||
from lerobot.teleoperators.utils import TeleopEvents
|
from lerobot.teleoperators.utils import TeleopEvents
|
||||||
@@ -83,7 +84,6 @@ from lerobot.utils.constants import (
|
|||||||
PRETRAINED_MODEL_DIR,
|
PRETRAINED_MODEL_DIR,
|
||||||
TRAINING_STATE_DIR,
|
TRAINING_STATE_DIR,
|
||||||
)
|
)
|
||||||
from lerobot.utils.process import ProcessSignalHandler
|
|
||||||
from lerobot.utils.random_utils import set_seed
|
from lerobot.utils.random_utils import set_seed
|
||||||
from lerobot.utils.train_utils import (
|
from lerobot.utils.train_utils import (
|
||||||
get_step_checkpoint_dir,
|
get_step_checkpoint_dir,
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ import pytest
|
|||||||
from lerobot.motors import Motor, MotorCalibration, MotorNormMode
|
from lerobot.motors import Motor, MotorCalibration, MotorNormMode
|
||||||
from lerobot.motors.dynamixel import MODEL_NUMBER_TABLE, DynamixelMotorsBus
|
from lerobot.motors.dynamixel import MODEL_NUMBER_TABLE, DynamixelMotorsBus
|
||||||
from lerobot.motors.dynamixel.tables import X_SERIES_CONTROL_TABLE
|
from lerobot.motors.dynamixel.tables import X_SERIES_CONTROL_TABLE
|
||||||
from lerobot.utils.encoding_utils import encode_twos_complement
|
from lerobot.motors.encoding_utils import encode_twos_complement
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import dynamixel_sdk as dxl
|
import dynamixel_sdk as dxl
|
||||||
|
|||||||
@@ -22,9 +22,9 @@ from unittest.mock import MagicMock, patch
|
|||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from lerobot.motors import Motor, MotorCalibration, MotorNormMode
|
from lerobot.motors import Motor, MotorCalibration, MotorNormMode
|
||||||
|
from lerobot.motors.encoding_utils import encode_sign_magnitude
|
||||||
from lerobot.motors.feetech import MODEL_NUMBER, MODEL_NUMBER_TABLE, FeetechMotorsBus
|
from lerobot.motors.feetech import MODEL_NUMBER, MODEL_NUMBER_TABLE, FeetechMotorsBus
|
||||||
from lerobot.motors.feetech.tables import STS_SMS_SERIES_CONTROL_TABLE
|
from lerobot.motors.feetech.tables import STS_SMS_SERIES_CONTROL_TABLE
|
||||||
from lerobot.utils.encoding_utils import encode_sign_magnitude
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import scservo_sdk as scs
|
import scservo_sdk as scs
|
||||||
|
|||||||
@@ -16,7 +16,7 @@
|
|||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from lerobot.utils.encoding_utils import (
|
from lerobot.motors.encoding_utils import (
|
||||||
decode_sign_magnitude,
|
decode_sign_magnitude,
|
||||||
decode_twos_complement,
|
decode_twos_complement,
|
||||||
encode_sign_magnitude,
|
encode_sign_magnitude,
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ from unittest.mock import patch
|
|||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from lerobot.utils.process import ProcessSignalHandler
|
from lerobot.rl.process import ProcessSignalHandler
|
||||||
|
|
||||||
|
|
||||||
# Fixture to reset shutdown_event_counter and original signal handlers before and after each test
|
# Fixture to reset shutdown_event_counter and original signal handlers before and after each test
|
||||||
|
|||||||
Reference in New Issue
Block a user