minor fix

uzh-rpg · Mar 15, 2022 · 109295e · 109295e
1 parent 80b256c
commit 109295e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/envtest/python/run_vision_demo.py b/envtest/python/run_vision_demo.py
@@ -7,7 +7,7 @@
 import cv2
 import numpy as np
 from flightgym import VisionEnv_v1
-from flightrl.rpg_baselines.torch.envs import vec_env_wrapper as wrapper
+from rpg_baselines.torch.envs import vec_env_wrapper as wrapper
 from ruamel.yaml import YAML, RoundTripDumper, dump
 
 

diff --git a/envtest/python/run_vision_ppo.py b/envtest/python/run_vision_ppo.py
@@ -11,9 +11,9 @@
 from stable_baselines3.common.utils import get_device
 from stable_baselines3.ppo.policies import MlpPolicy
 
-from flightrl.rpg_baselines.torch.common.ppo import PPO
-from flightrl.rpg_baselines.torch.envs import vec_env_wrapper as wrapper
-from flightrl.rpg_baselines.torch.common.util import test_policy
+from rpg_baselines.torch.common.ppo import PPO
+from rpg_baselines.torch.envs import vec_env_wrapper as wrapper
+from rpg_baselines.torch.common.util import test_policy
 
 
 def configure_random_seed(seed, env=None):

diff --git a/setup_py.bash b/setup_py.bash
@@ -40,6 +40,7 @@ rm $project_path/flightmare/flightrender/RPG_Flightmare.zip
 
 # 
 echo "export FLIGHTMARE_PATH=$project_path/flightmare" >> ~/.bashrc
+source ~/.bashrc
 
 # 
 echo "Createing an conda environment from the environment.yaml file. Make sure you have anaconda installed"
-Original file line number
+Diff line change
@@ Expand Up / @@ -40,6 +40,7 @@ rm $project_path/flightmare/flightrender/RPG_Flightmare.zip @@
     #
     echo "export FLIGHTMARE_PATH=$project_path/flightmare" >> ~/.bashrc
+    source ~/.bashrc
     #
     echo "Createing an conda environment from the environment.yaml file. Make sure you have anaconda installed"
@@ Expand Down @@