fixes in imports

ramos-ai · Feb 2, 2024 · a9e4e20 · a9e4e20
1 parent 4537724
commit a9e4e20
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,6 +15,9 @@ __pycache__/
 # Backup
 backup
 
+# Results
+results
+
 # Results
 .ipynb_checkpoints/
 analytics

diff --git a/experiments/gtq_learning_exp.py b/experiments/gtq_learning_exp.py
@@ -7,7 +7,7 @@
 from route_choice_env.agents.gtq_learning import GTQLearning
 from route_choice_env.policy import EpsilonGreedy
 
-from experiments.experiment import Experiment
+from experiment import Experiment
 
 
 N_EXPERIMENTS = 0

diff --git a/experiments/main.py b/experiments/main.py
@@ -9,11 +9,11 @@
 import numpy as np
 from pathlib import Path
 
-from experiments.experiment import Experiment
+from experiment import Experiment
 
-from experiments.rmq_learning_exp import RMQLearningExperiment
-from experiments.tq_learning_exp import TQLearningExperiment
-from experiments.gtq_learning_exp import GTQLearningExperiment
+from rmq_learning_exp import RMQLearningExperiment
+from tq_learning_exp import TQLearningExperiment
+from gtq_learning_exp import GTQLearningExperiment
 
 
 class InvalidAlgorithm(Exception):

diff --git a/experiments/rmq_learning_exp.py b/experiments/rmq_learning_exp.py
@@ -7,7 +7,7 @@
 from route_choice_env.agents.rmq_learning import RMQLearning
 from route_choice_env.policy import EpsilonGreedy
 
-from experiments.experiment import Experiment
+from experiment import Experiment
 
 
 def get_rmq_learning_agents(env: RouteChoicePZ, policy: Policy):

diff --git a/experiments/tq_learning_exp.py b/experiments/tq_learning_exp.py
@@ -7,7 +7,7 @@
 from route_choice_env.agents.tq_learning import TQLearning
 from route_choice_env.policy import EpsilonGreedy
 
-from experiments.experiment import Experiment
+from experiment import Experiment
 
 
 N_EXPERIMENTS = 0