Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/lerobot/cameras/zmq/image_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@

from lerobot.cameras.configs import ColorMode
from lerobot.cameras.opencv import OpenCVCamera, OpenCVCameraConfig
from lerobot.utils.utils import init_logging

logger = logging.getLogger(__name__)

Expand Down Expand Up @@ -177,6 +178,6 @@ def run(self):


if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
init_logging()
config = {"fps": 30, "cameras": {"head_camera": {"device_id": 4, "shape": [480, 640]}}}
ImageServer(config, port=5555).run()
3 changes: 2 additions & 1 deletion src/lerobot/rl/eval_policy.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,10 +29,11 @@
gamepad, # noqa: F401
so_leader, # noqa: F401
)
from lerobot.utils.utils import init_logging

from .gym_manipulator import make_robot_env

logging.basicConfig(level=logging.INFO)
init_logging()


def eval_policy(env, policy, n_episodes):
Expand Down
4 changes: 2 additions & 2 deletions src/lerobot/rl/gym_manipulator.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,11 +75,11 @@
from lerobot.teleoperators.utils import TeleopEvents
from lerobot.utils.constants import ACTION, DONE, OBS_IMAGES, OBS_STATE, REWARD
from lerobot.utils.robot_utils import precise_sleep
from lerobot.utils.utils import log_say
from lerobot.utils.utils import init_logging, log_say

from .joint_observations_processor import JointVelocityProcessorStep, MotorCurrentProcessorStep

logging.basicConfig(level=logging.INFO)
init_logging()


@dataclass
Expand Down