From 0c41675986f98578954fe635a5200cc757223ff8 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Sun, 24 Mar 2024 17:57:01 +0100
Subject: [PATCH] fix __init__ import Base

---
 lerobot/common/envs/simxarm/simxarm/__init__.py | 1 +
 pyproject.toml                                  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/lerobot/common/envs/simxarm/simxarm/__init__.py b/lerobot/common/envs/simxarm/simxarm/__init__.py
index 870eb5a9..9906ed8d 100644
--- a/lerobot/common/envs/simxarm/simxarm/__init__.py
+++ b/lerobot/common/envs/simxarm/simxarm/__init__.py
@@ -4,6 +4,7 @@ import gym
 import numpy as np
 from gym.wrappers import TimeLimit
 
+from lerobot.common.envs.simxarm.simxarm.task.base import Base as Base
 from lerobot.common.envs.simxarm.simxarm.task.lift import Lift
 from lerobot.common.envs.simxarm.simxarm.task.peg_in_box import PegInBox
 from lerobot.common.envs.simxarm.simxarm.task.push import Push
diff --git a/pyproject.toml b/pyproject.toml
index 5f6c9456..d1cdd055 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -89,7 +89,7 @@ exclude = [
 
 [tool.ruff.lint]
 select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
-
+ignore-init-module-imports = true
 
 [tool.poetry-dynamic-versioning]
 enable = true