Fix imports

2021-09-08 20:45:45 +02:00
parent 3a6139286d
commit 5ff4b42c0e
4 changed files with 7 additions and 7 deletions
--- a/scratch/etienne/intersimple/ppo_speed_image_lowres.py
+++ b/scratch/etienne/intersimple/ppo_speed_image_lowres.py
@@ -1,6 +1,6 @@
 # %%
 from stable_baselines3 import PPO
-from intersimple.intersimple import NRasterized, speed_reward
+from intersim.envs.intersimple import NRasterized, speed_reward
 import functools

 model_name = "ppo_speed_image_lowres"