From 81420876498c21788ddfc786a5fdbea5b8138211 Mon Sep 17 00:00:00 2001 From: Lu Junjie Date: Sat, 2 Nov 2024 12:25:00 +0800 Subject: [PATCH] fix a warning issue and update readme --- README.md | 2 +- flightpolicy/envs/vec_env_wrapper.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 4c0706a..28c5698 100644 --- a/README.md +++ b/README.md @@ -78,7 +78,7 @@ conda activate yopo conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia pip install opencv-python -pip install gym stable-baselines3==1.5.0 +pip install gym==0.21.0 stable-baselines3==1.5.0 pip install scipy scikit-build ruamel-yaml==0.17.21 numpy==1.22.3 tensorboard==2.9.1 empy catkin_pkg ``` **5. build the flightlib** diff --git a/flightpolicy/envs/vec_env_wrapper.py b/flightpolicy/envs/vec_env_wrapper.py index 7840a11..4047833 100644 --- a/flightpolicy/envs/vec_env_wrapper.py +++ b/flightpolicy/envs/vec_env_wrapper.py @@ -34,7 +34,7 @@ class FlightEnvVec(VecEnv): self._gray_img_obs = np.zeros([self.num_envs, self.img_width * self.img_height], dtype=np.uint8) self._depth_img_obs = np.zeros([self.num_envs, self.img_width * self.img_height], dtype=np.float32) self._reward = np.zeros([self.num_envs, self.reward_dim], dtype=np.float32) - self._done = np.zeros((self.num_envs), dtype=np.bool) + self._done = np.zeros((self.num_envs), dtype=bool) self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # observation: [p_wb, v_b, a_b, q_wb] (in Body Frame); action: dp_pred; reward: cost