Skip to content

Commit

Permalink
Fix annotation issue
Browse files Browse the repository at this point in the history
  • Loading branch information
xkiixkii committed Oct 4, 2024
1 parent dd02667 commit 1866d73
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
12 changes: 3 additions & 9 deletions deformable_gym/envs/mujoco/grasp_env.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from __future__ import annotations

from typing import Any, Dict, Tuple

import mujoco
Expand All @@ -14,14 +16,6 @@ class GraspEnv(BaseMJEnv):
A custom MuJoCo environment for a grasping task, where a robot attempts to grasp an object.
"""

meta_data = {
"render_mode": [
"human",
"rgb_array",
"depth_array",
],
}

def __init__(
self,
robot_name: str,
Expand All @@ -30,7 +24,7 @@ def __init__(
observable_object_pos: bool = True,
control_type: str = "mocap",
max_sim_time: float = 6,
render_mode: str | None = "human",
render_mode: str | None = None,
mocap_cfg: Dict[str, str] | None = None,
init_frame: str | None = None,
default_cam_config: Dict[str, Any] | None = None,
Expand Down
2 changes: 2 additions & 0 deletions deformable_gym/helpers/mj_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from __future__ import annotations

from dataclasses import dataclass
from typing import Any, Dict, List, Tuple

Expand Down
2 changes: 2 additions & 0 deletions deformable_gym/robots/mj_robot.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from __future__ import annotations

from abc import ABC
from typing import List

Expand Down

0 comments on commit 1866d73

Please sign in to comment.