fix path configs

Author: Vindulamj

code/ddpg_training.py

@@ -35,9 +35,9 @@
 
 parser = argparse.ArgumentParser(description='Model arguments')
 parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
-parser.add_argument('--intersection_dir', default='/home/gridsan/chickert/vindula/scenarioenv/dataset/salt-lake-city', type=str, help='Path to intersection dataset')
+parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
 parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name')
-parser.add_argument('--wandb_entity', default='vindula', type=str, help='Weights and biases entity name')
+parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
 parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
 parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
 parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -49,6 +49,8 @@
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 wandb.init(project=args.wandb_project, entity=args.wandb_entity)
 
 ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)

code/env_demo.py

@@ -8,14 +8,16 @@
 from sumo.constants import REGULAR
 
 parser = argparse.ArgumentParser(description='Demo run arguments')
-parser.add_argument('--dir', default='/Users/bfreydt/MIT_local/IntersectionZoo/wd/new_exp', type=str, help='Result directory')
-parser.add_argument('--intersection_dir', default='/Users/bfreydt/MIT_local/IntersectionZoo/dataset/salt-lake-city', type=str, help='Path to intersection dataset')
+parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
+parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
 parser.add_argument('--penetration', default=0.33, type=str, help='Eco drive adoption rate')
 parser.add_argument('--temperature_humidity', default='68_46', type=str, help='Temperature and humidity for evaluations')
 
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 tasks = PathTaskContext(
     dir=Path(args.intersection_dir),                    
     single_approach=True,
@@ -29,6 +31,7 @@
     working_dir=Path(args.dir),
     moves_emissions_models=[args.temperature_humidity],
     fleet_reward_ratio=1,
+    visualize_sumo=True,
 )
 
 # Create the environment
@@ -64,7 +67,6 @@ def filter_rew(rew: dict):
     # Print the observations and reward
     print("Observations:", filter_obs(obs))
     print("Reward:", filter_rew(reward))
-    input("Press Enter to continue...")
 
 # Close the environment
 env.close()

code/policy_evaluation.py

@@ -30,10 +30,10 @@
 from sumo.constants import REGULAR
 
 parser = argparse.ArgumentParser(description='Model arguments')
-parser.add_argument('--dir', default='/home/gridsan/vindula/neurips24/scenarioenv', type=str, help='Result directory')
-parser.add_argument('--intersection_dir', default='/home/gridsan/vindula/neurips24/scenarioenv/dataset/test', type=str, help='Path to intersection dataset')
+parser.add_argument('--dir', default='wd/ppo_run', type=str, help='Result directory')
+parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
 parser.add_argument('--num_workers', default=5, type=str, help='Number of workers')
-parser.add_argument('--checkpoint', default='/home/gridsan/vindula/neurips24/scenarioenv/wd/ppo_run/runs/0/20240530_2354', type=str, help='Checkpoint path')
+parser.add_argument('--checkpoint', default='/wd/ppo_run/runs/0/20240530_2354', type=str, help='Checkpoint path')
 parser.add_argument('--eval_per_task', default=3, type=str, help='How many times to evaluate each task')
 
 parser.add_argument('--penetration', default=1.0, type=str, help='Eco drive adoption rate')
@@ -42,6 +42,8 @@
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)
 
 tasks = PathTaskContext(

code/ppo_training.py

@@ -35,9 +35,9 @@
 
 parser = argparse.ArgumentParser(description='Model arguments')
 parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
-parser.add_argument('--intersection_dir', default='/home/gridsan/vindula/scenarioenv/dataset/salt-lake-city', type=str, help='Path to intersection dataset')
+parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
 parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name')
-parser.add_argument('--wandb_entity', default='vindula', type=str, help='Weights and biases entity name')
+parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
 parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
 parser.add_argument('--num_gpus', default=1, type=str, help='Number of GPUs')
 parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -49,6 +49,8 @@
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 wandb.init(project=args.wandb_project, entity=args.wandb_entity)
 
 ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)

code/synthetic_ddpg_training.py

@@ -36,7 +36,7 @@
 parser = argparse.ArgumentParser(description='Model arguments')
 parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
 parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name')
-parser.add_argument('--wandb_entity', default='vindula', type=str, help='Weights and biases entity name')
+parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
 parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
 parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
 parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -48,6 +48,8 @@
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 wandb.init(project=args.wandb_project, entity=args.wandb_entity)
 
 ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)

code/synthetic_ppo_training.py

@@ -36,7 +36,7 @@
 parser = argparse.ArgumentParser(description='Model arguments')
 parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
 parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name')
-parser.add_argument('--wandb_entity', default='vindula', type=str, help='Weights and biases entity name')
+parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
 parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
 parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
 parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -48,6 +48,8 @@
 args = parser.parse_args()
 print(args)
 
+Path(args.dir).mkdir(parents=True, exist_ok=True)
+
 wandb.init(project=args.wandb_project, entity=args.wandb_entity)
 
 ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)