This commit is contained in:
Philipp Wu 2024-04-04 16:41:57 -07:00
parent abd53a4c2b
commit 16e09073bf
5 changed files with 15 additions and 9 deletions

View file

@ -2,26 +2,26 @@ repos:
# remove unused python imports # remove unused python imports
- repo: https://github.com/myint/autoflake.git - repo: https://github.com/myint/autoflake.git
rev: v2.1.1 rev: v2.3.1
hooks: hooks:
- id: autoflake - id: autoflake
args: ["--in-place", "--remove-all-unused-imports", "--ignore-init-module-imports"] args: ["--in-place", "--remove-all-unused-imports", "--ignore-init-module-imports"]
# sort imports # sort imports
- repo: https://github.com/timothycrosley/isort - repo: https://github.com/timothycrosley/isort
rev: 5.12.0 rev: 5.13.2
hooks: hooks:
- id: isort - id: isort
# code format according to black # code format according to black
- repo: https://github.com/ambv/black - repo: https://github.com/ambv/black
rev: 24.2.0 rev: 24.3.0
hooks: hooks:
- id: black - id: black
# check for python styling with flake8 # check for python styling with flake8
- repo: https://github.com/PyCQA/flake8 - repo: https://github.com/PyCQA/flake8
rev: 6.0.0 rev: 7.0.0
hooks: hooks:
- id: flake8 - id: flake8
additional_dependencies: [ additional_dependencies: [
@ -33,6 +33,6 @@ repos:
# cleanup notebooks # cleanup notebooks
- repo: https://github.com/kynan/nbstripout - repo: https://github.com/kynan/nbstripout
rev: 0.6.1 rev: 0.7.1
hooks: hooks:
- id: nbstripout - id: nbstripout

View file

@ -1,4 +1,5 @@
"""Manipulator composer class.""" """Manipulator composer class."""
import abc import abc
from pathlib import Path from pathlib import Path
from typing import Dict, List, Optional, Tuple, Union from typing import Dict, List, Optional, Tuple, Union

View file

@ -1,4 +1,5 @@
"""Simple floor arenas.""" """Simple floor arenas."""
from typing import Tuple from typing import Tuple
import numpy as np import numpy as np

View file

@ -1,4 +1,5 @@
"""A task where a walker must learn to stand.""" """A task where a walker must learn to stand."""
from typing import Optional from typing import Optional
import numpy as np import numpy as np

View file

@ -86,11 +86,14 @@ class DynamixelRobot(Robot):
if gripper_config is not None: if gripper_config is not None:
current_joints = current_joints[:-1] current_joints = current_joints[:-1]
start_joints = start_joints[:-1] start_joints = start_joints[:-1]
for idx, (c_joint, s_joint, joint_offset) in enumerate(zip( for idx, (c_joint, s_joint, joint_offset) in enumerate(
current_joints, start_joints, self._joint_offsets zip(current_joints, start_joints, self._joint_offsets)
)): ):
new_joint_offsets.append( new_joint_offsets.append(
np.pi * 2 * np.round((-s_joint + c_joint) / (2 * np.pi)) * self._joint_signs[idx] np.pi
* 2
* np.round((-s_joint + c_joint) / (2 * np.pi))
* self._joint_signs[idx]
+ joint_offset + joint_offset
) )
if gripper_config is not None: if gripper_config is not None: