Merge pull request #10 from CLAIR-LAB-TECHNION/main

merge
CLAIR-LAB-TECHNION · Jul 15, 2024 · 4461baf · 4461baf
2 parents 41fd5cb + 3a5b028
commit 4461baf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 24 deletions.
diff --git a/tutorials/dqn_tutorial/dqn/__init__.py b/tutorials/dqn_tutorial/dqn/__init__.py
@@ -1,11 +1 @@
-from dqn_tutorial.dqn.collect_data import collect_one_step, epsilon_greedy_action_selection, linear_schedule
-from dqn_tutorial.dqn.q_network import QNetwork
-from dqn_tutorial.dqn.replay_buffer import ReplayBuffer
 
-__all__ = [
-    "QNetwork",
-    "ReplayBuffer",
-    "collect_one_step",
-    "linear_schedule",
-    "epsilon_greedy_action_selection",
-]
diff --git a/tutorials/dqn_tutorial/dqn/collect_data.py b/tutorials/dqn_tutorial/dqn/collect_data.py
@@ -3,8 +3,8 @@
 import torch as th
 from gymnasium import spaces
 
-from dqn_tutorial.dqn.q_network import QNetwork
-from dqn_tutorial.dqn.replay_buffer import ReplayBuffer
+from dqn.q_network import QNetwork
+from dqn.replay_buffer import ReplayBuffer
 
 
 def epsilon_greedy_action_selection(

diff --git a/tutorials/dqn_tutorial/dqn/evaluation.py b/tutorials/dqn_tutorial/dqn/evaluation.py
@@ -6,8 +6,8 @@
 from gymnasium import spaces
 from gymnasium.wrappers.monitoring.video_recorder import VideoRecorder
 
-from dqn_tutorial.dqn.collect_data import epsilon_greedy_action_selection
-from dqn_tutorial.dqn.q_network import QNetwork
+from dqn.collect_data import epsilon_greedy_action_selection
+from dqn.q_network import QNetwork
 
 
 def evaluate_policy(

diff --git a/tutorials/dqn_tutorial/fqi/__init__.py b/tutorials/dqn_tutorial/fqi/__init__.py
@@ -1,11 +1 @@
-from dqn_tutorial.fqi.collect_data import collect_data, load_data, save_data
-from dqn_tutorial.fqi.fqi import create_model_input, evaluate, get_q_values
 
-__all__ = [
-    "collect_data",
-    "load_data",
-    "save_data",
-    "create_model_input",
-    "evaluate",
-    "get_q_values",
-]