minor fix

Author: yun-long

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
 
 

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):

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"