gym-examples 3.0.10__py3-none-any.whl → 3.0.12__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.10"
+__version__ = "3.0.12"

gym_examples/envs/wsn_env.py

@@ -83,6 +83,12 @@ class WSNRoutingEnv(gym.Env):
         self.packet_latency = np.zeros(self.n_sensors)  # Latency for each packet
         self.total_packets_sent_by_sensors = 0
 
+        self.network_throughput = None
+        self.energy_efficiency = None
+        self.packet_delivery_ratio = None
+        self.network_lifetime = None
+        self.average_latency = None
+
         return self._get_obs()
 
     def step(self, actions):
@@ -167,14 +173,19 @@ class WSNRoutingEnv(gym.Env):
         if self.first_node_dead_time is None and np.any(self.remaining_energy <= 0):
             self.first_node_dead_time = self.steps
 
+        self.get_metrics()
+
         return self._get_obs(), rewards, dones, {}
 
     def _get_obs(self):
+
+        performance = self.get_metrics()
+
         return [{'remaining_energy': np.array([e]), 
                  'consumption_energy': np.array([initial_energy - e]),
                  'sensor_positions': p,
-                 'number_of_packets': np.array([d]),
-                 'statistics': self.get_metrics()} for e, p, d in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)]
+                 'number_of_packets': np.array([d])
+                } for e, p, d in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)]
 
     def _get_observation_space(self):
         return Dict({
@@ -330,20 +341,20 @@ class WSNRoutingEnv(gym.Env):
     
     def get_metrics(self):
         # Calculate network throughput
-        network_throughput = self.packets_delivered / self.steps if self.steps > 0 else 0
+        self.network_throughput = self.packets_delivered / self.steps if self.steps > 0 else 0
         # Calculate energy efficiency
-        energy_efficiency = self.packets_delivered / self.total_energy_consumed if self.total_energy_consumed > 0 else 0
+        self.energy_efficiency = self.packets_delivered / self.total_energy_consumed if self.total_energy_consumed > 0 else 0
         # Calculate packet delivery ratio
-        packet_delivery_ratio = self.packets_delivered / self.total_packets_sent_by_sensors if self.total_packets_sent_by_sensors > 0 else 0
+        self.packet_delivery_ratio = self.packets_delivered / self.total_packets_sent_by_sensors if self.total_packets_sent_by_sensors > 0 else 0
         # Calculate network lifetime
-        network_lifetime = self.first_node_dead_time if self.first_node_dead_time is not None else self.steps
+        self.network_lifetime = self.first_node_dead_time if self.first_node_dead_time is not None else self.steps
         # Calculate average latency
-        average_latency = self.total_latency / self.packets_delivered if self.packets_delivered > 0 else 0
+        self.average_latency = self.total_latency / self.packets_delivered if self.packets_delivered > 0 else 0
 
         return {
-            "network_throughput": network_throughput,
-            "energy_efficiency": energy_efficiency,
-            "packet_delivery_ratio": packet_delivery_ratio,
-            "network_lifetime": network_lifetime,
-            "average_latency": average_latency
+            "network_throughput": self.network_throughput,
+            "energy_efficiency": self.energy_efficiency,
+            "packet_delivery_ratio": self.packet_delivery_ratio,
+            "network_lifetime": self.network_lifetime,
+            "average_latency": self.average_latency
         }

{gym_examples-3.0.10.dist-info → gym_examples-3.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gym-examples
-Version: 3.0.10
+Version: 3.0.12
 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.12.dist-info/RECORD

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=kcZhlTThwzjMoXcErF9Rpjo73RBaXYxCJXEKmNW1Stw,193
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=LcfTSNY6AKL-gW5adcmS9T2aOxerBuFPl8AVGaeRH50,17904
+gym_examples-3.0.12.dist-info/METADATA,sha256=nT5ITiPaKcPghJtG2m_47AHIE7f6olO5pVxB-xPNGNQ,411
+gym_examples-3.0.12.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.12.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.12.dist-info/RECORD,,

gym_examples-3.0.10.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=RBpA4jzbyv6IceDsSXiR6P-U6Aj3OvlZ7eCOQr7F7j0,193
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=xg4nJw0EyK3_Vt4vJpyhEIG44FnpFPSOiSsQM1KxDt8,17613
-gym_examples-3.0.10.dist-info/METADATA,sha256=xmZDjTODWMEFRTJHb9bQPAzvMw9NP7yYRWFIfFKqQCA,411
-gym_examples-3.0.10.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.10.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.10.dist-info/RECORD,,