gym-examples 3.0.180__py3-none-any.whl → 3.0.182__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.180"
+__version__ = "3.0.182"

gym_examples/envs/wsn_env.py

@@ -52,11 +52,13 @@ class WSNRoutingEnv(gym.Env):
 
     print_stats = False  # Global flag to control printing of statistics
 
-    def __init__(self, n_sensors = 20, coverage_radius=(upper_bound - lower_bound)/4):
+    def __init__(self, n_sensors = 20, coverage_radius=(upper_bound - lower_bound)/4, num_timesteps = None, version = None):
 
         super(WSNRoutingEnv, self).__init__()
 
         # Initialize list of episode metrics
+        self.num_timesteps = num_timesteps # This argument is for the PPO algorithm
+        self.version = version # This argument is for the PPO algorithm
         self.number_of_steps = 0 # Total number of steps taken by the agent since the beginning of the training 
         self.episode_returns = []
         self.episode_std_remaining_energy = []
@@ -212,7 +214,7 @@ class WSNRoutingEnv(gym.Env):
         if os.getenv('PRINT_STATS') == 'True': # We are trying to extract only the statistics for the PPO algorithm
             self.number_of_steps += 1
             self.episode_return += rewards
-            if self.number_of_steps >= num_timesteps:
+            if self.number_of_steps >= self.num_timesteps:
                 self.episode_returns.append(self.episode_return)
                 self.episode_std_remaining_energy.append(np.std(self.remaining_energy))
                 self.episode_mean_remaining_energy.append(np.mean(self.remaining_energy))
@@ -222,14 +224,14 @@ class WSNRoutingEnv(gym.Env):
                 self.episode_network_lifetime.append(self.network_lifetime)
                 self.episode_average_latency.append(self.average_latency)
 
-                np.save(f"{base_back_up_dir}returns_QMIX_{version}.npy", np.array(self.episode_returns))
-                np.save(f"{base_back_up_dir}std_remaining_energy_QMIX_{version}.npy", np.array(self.episode_std_remaining_energy))
-                np.save(f"{base_back_up_dir}total_consumption_energy_QMIX_{version}.npy", np.array(self.episode_total_consumption_energy))
-                np.save(f"{base_back_up_dir}mean_remaining_energy_QMIX_{version}.npy", np.array(self.episode_mean_remaining_energy))
-                np.save(f"{base_back_up_dir}network_throughput_QMIX_{version}.npy", np.array(self.episode_network_throughput))
-                np.save(f"{base_back_up_dir}packet_delivery_ratio_QMIX_{version}.npy", np.array(self.episode_packet_delivery_ratio))
-                np.save(f"{base_back_up_dir}network_lifetime_QMIX_{version}.npy", np.array(self.episode_network_lifetime))
-                np.save(f"{base_back_up_dir}average_latency_QMIX_{version}.npy", np.array(self.episode_average_latency))
+                np.save(f"{base_back_up_dir}returns_QMIX_{self.version}.npy", np.array(self.episode_returns))
+                np.save(f"{base_back_up_dir}std_remaining_energy_QMIX_{self.version}.npy", np.array(self.episode_std_remaining_energy))
+                np.save(f"{base_back_up_dir}total_consumption_energy_QMIX_{self.version}.npy", np.array(self.episode_total_consumption_energy))
+                np.save(f"{base_back_up_dir}mean_remaining_energy_QMIX_{self.version}.npy", np.array(self.episode_mean_remaining_energy))
+                np.save(f"{base_back_up_dir}network_throughput_QMIX_{self.version}.npy", np.array(self.episode_network_throughput))
+                np.save(f"{base_back_up_dir}packet_delivery_ratio_QMIX_{self.version}.npy", np.array(self.episode_packet_delivery_ratio))
+                np.save(f"{base_back_up_dir}network_lifetime_QMIX_{self.version}.npy", np.array(self.episode_network_lifetime))
+                np.save(f"{base_back_up_dir}average_latency_QMIX_{self.version}.npy", np.array(self.episode_average_latency))
 
         return self._get_obs(), rewards, dones, {}
 

{gym_examples-3.0.180.dist-info → gym_examples-3.0.182.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gym-examples
-Version: 3.0.180
+Version: 3.0.182
 Summary: A custom environment for multi-agent reinforcement learning focused on WSN routing.
 Home-page: https://github.com/gedji/CODES.git
 Author: Georges Djimefo

gym_examples-3.0.182.dist-info/RECORD

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=RiS3AduI_PATwPjqyEMvdXeVB7NUbBrYUWBJchVfwuw,194
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=vJ94e6nzg1iSxSyfEYffTjhN7szbIz1ZaqU4tG_U0_Y,25901
+gym_examples-3.0.182.dist-info/METADATA,sha256=R7BDOPdQSi9IwPOQGfq9a8uqNBl0hYvwMHjJD2l-Jo8,412
+gym_examples-3.0.182.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.182.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.182.dist-info/RECORD,,

gym_examples-3.0.180.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=u596NUGuugLKs4xG38CdXc1ChDTjdSFE4qwRq07dSco,194
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=xXwNW7X-pFQT0T_vJwI7WIkoAhUhtmz7nTEokL9d30Q,25660
-gym_examples-3.0.180.dist-info/METADATA,sha256=esLOv32LCP3zP5pOa-zczAQTsz0ZSha7VqVAL0pcsNg,412
-gym_examples-3.0.180.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.180.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.180.dist-info/RECORD,,