diff --git a/env.py b/env.py index b5a54c4..4dd4d68 100644 --- a/env.py +++ b/env.py @@ -124,7 +124,7 @@ class PartitionMazeEnv(gym.Env): for i in range(len(self.row_cuts) - 1): for j in range(len(self.col_cuts) - 1): d = (self.col_cuts[j+1] - self.col_cuts[j]) * self.W * \ - (self.row_cuts[i] + self.row_cuts[i+1]) * self.H + (self.row_cuts[i+1] - self.row_cuts[i]) * self.H rho_time_limit = (self.flight_time_factor - self.trans_time_factor) / \ (self.comp_time_factor - self.trans_time_factor) rho_energy_limit = (self.battery_energy_capacity - self.flight_energy_factor * d - self.trans_energy_factor * d) / \ diff --git a/human_action.py b/human_action.py new file mode 100644 index 0000000..92df03f --- /dev/null +++ b/human_action.py @@ -0,0 +1,15 @@ +from env import PartitionMazeEnv + +env = PartitionMazeEnv() + +state = env.reset() +print(state) + +action_series = [[0], [0], [0.5], [0], [0.1], [0.3]] + +for i in range(10): + action = action_series[i] + state, reward, done, info, _ = env.step(action) + print(state, reward, done, info) + if done: + break \ No newline at end of file