Skip to content
Snippets Groups Projects
Commit cd59797b authored by u214892's avatar u214892
Browse files

#42 run baselines in ci

parent 1a73719c
No related branches found
No related tags found
No related merge requests found
......@@ -6,7 +6,6 @@ from collections import deque
import matplotlib.pyplot as plt
import numpy as np
import torch
from dueling_double_dqn import Agent
from importlib_resources import path
import torch_training.Nets
......@@ -15,8 +14,11 @@ from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
from torch_training.dueling_double_dqn import Agent
from utils.observation_utils import norm_obs_clip, split_tree
print("multi_agent_trainging.py (1)")
def main(argv):
try:
......@@ -219,6 +221,10 @@ def main(argv):
plt.show()
print("multi_agent_trainging.py (2)")
if __name__ == '__main__':
print("main")
main(sys.argv[1:])
print("multi_agent_trainging.py (3)")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment