Add typos checks (#770)

tests/mock_cv2.py

@@ -18,11 +18,11 @@ def _generate_image(width: int, height: int):
     return np.random.randint(0, 256, size=(height, width, 3), dtype=np.uint8)
 
 
-def cvtColor(color_image, color_convertion):  # noqa: N802
-    if color_convertion in [COLOR_RGB2BGR, COLOR_BGR2RGB]:
+def cvtColor(color_image, color_conversion):  # noqa: N802
+    if color_conversion in [COLOR_RGB2BGR, COLOR_BGR2RGB]:
         return color_image[:, :, [2, 1, 0]]
     else:
-        raise NotImplementedError(color_convertion)
+        raise NotImplementedError(color_conversion)
 
 
 def rotate(color_image, rotation):

tests/test_cameras.py

@@ -27,7 +27,7 @@ import pytest
 from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
 from tests.utils import TEST_CAMERA_TYPES, make_camera, require_camera
 
-# Maximum absolute difference between two consecutive images recored by a camera.
+# Maximum absolute difference between two consecutive images recorded by a camera.
 # This value differs with respect to the camera.
 MAX_PIXEL_DIFFERENCE = 25
 

tests/test_control_robot.py

@@ -179,7 +179,7 @@ def test_record_and_replay_and_policy(tmp_path, request, robot_type, mock):
     policy.save_pretrained(pretrained_policy_path)
 
     # In `examples/9_use_aloha.md`, we advise using `num_image_writer_processes=1`
-    # during inference, to reach constent fps, so we test this here.
+    # during inference, to reach constant fps, so we test this here.
     if robot_type == "aloha":
         num_image_writer_processes = 1
 

tests/test_datasets.py

@@ -486,7 +486,7 @@ def test_backward_compatibility(repo_id):
         old_frame = load_file(test_dir / f"frame_{i}.safetensors")  # noqa: B023
 
         # ignore language instructions (if exists) in language conditioned datasets
-        # TODO (michel-aractingi): transform language obs to langauge embeddings via tokenizer
+        # TODO (michel-aractingi): transform language obs to language embeddings via tokenizer
         new_frame.pop("language_instruction", None)
         old_frame.pop("language_instruction", None)
         new_frame.pop("task", None)

tests/test_delta_timestamps.py

@@ -32,10 +32,10 @@ def calculate_episode_data_index(hf_dataset: datasets.Dataset) -> dict[str, np.n
         ep_table = table.filter(pc.equal(table["episode_index"], ep_idx))
         episode_lengths.insert(ep_idx, len(ep_table))
 
-    cumulative_lenghts = list(accumulate(episode_lengths))
+    cumulative_lengths = list(accumulate(episode_lengths))
     return {
-        "from": np.array([0] + cumulative_lenghts[:-1], dtype=np.int64),
-        "to": np.array(cumulative_lenghts, dtype=np.int64),
+        "from": np.array([0] + cumulative_lengths[:-1], dtype=np.int64),
+        "to": np.array(cumulative_lengths, dtype=np.int64),
     }
 
 

tests/test_motors.py

@@ -88,7 +88,7 @@ def test_motors_bus(request, motor_type, mock):
 
     motors_bus = make_motors_bus(motor_type, mock=mock)
 
-    # Test reading and writting before connecting raises an error
+    # Test reading and writing before connecting raises an error
     with pytest.raises(RobotDeviceNotConnectedError):
         motors_bus.read("Torque_Enable")
     with pytest.raises(RobotDeviceNotConnectedError):

tests/test_robots.py

@@ -86,7 +86,7 @@ def test_robot(tmp_path, request, robot_type, mock):
     robot.connect()
     robot.teleop_step()
 
-    # Test data recorded during teleop are well formated
+    # Test data recorded during teleop are well formatted
     observation, action = robot.teleop_step(record_data=True)
     # State
     assert "observation.state" in observation