Fix linter/mypy issues

Author: araffin

dqn_tutorial/dqn/__init__.py

@@ -6,6 +6,6 @@
     "QNetwork",
     "ReplayBuffer",
     "collect_one_step",
-    "linear_schedule",
     "epsilon_greedy_action_selection",
+    "linear_schedule",
 ]

dqn_tutorial/fqi/__init__.py

@@ -3,9 +3,9 @@
 
 __all__ = [
     "collect_data",
-    "load_data",
-    "save_data",
     "create_model_input",
     "evaluate",
     "get_q_values",
+    "load_data",
+    "save_data",
 ]

dqn_tutorial/fqi/collect_data.py

@@ -44,7 +44,7 @@ def collect_data(env_id: str, n_steps: int = 50_000) -> OfflineData:
     for idx in range(n_steps):
         # Sample a random action
         action = env.action_space.sample()
-        next_obs, reward, terminated, truncated, info_ = env.step(action)
+        next_obs, reward, terminated, truncated, _info = env.step(action)
 
         # Store the transition
         observations[idx, :] = obs
@@ -80,7 +80,7 @@ def save_data(data: OfflineData, path: Path) -> None:
 
     np.savez(
         path,
-        **dict(
+        **dict(  # type: ignore[arg-type]
             observations=data.observations,
             next_observations=data.next_observations,
             rewards=data.rewards,