gym-examples 3.0.765__py3-none-any.whl → 3.0.767__py3-none-any.whl

gym_examples/__init__.py

@@ -5,4 +5,4 @@ register(
      entry_point="gym_examples.envs:WSNRoutingEnv",
 )
 
-__version__ = "3.0.765"
+__version__ = "3.0.767"

gym_examples/envs/wsn_env.py

@@ -167,7 +167,7 @@ class WSNRoutingEnv(gym.Env):
                 # Calculate the energy consumption for transmitting data to the base station
                 transmission_energy = self.transmission_energy(self.number_of_packets[i], self.distance_to_base[i])
                 if self.remaining_energy[i] < transmission_energy:
-                    self.remaining_energy[i] = 0
+                    # self.remaining_energy[i] = 0
                     continue  # Skip if the sensor does not have enough energy to transmit data to the base station
                 
                 self.update_sensor_energies(i, transmission_energy)
@@ -179,6 +179,8 @@ class WSNRoutingEnv(gym.Env):
                 self.total_latency += self.packet_latency[i] + latency_per_hop
                 self.packet_latency[i] = 0
 
+                self.number_of_packets[i] = 0 # Reset the number of packets of the sensor i
+
                 rewards[i] = np.array([max_reward] * input_dim) # Reward for transmitting data to the base station
                 dones[i] = True
             else:
@@ -189,19 +191,24 @@ class WSNRoutingEnv(gym.Env):
                 transmission_energy = self.transmission_energy(self.number_of_packets[i], distance)
                 reception_energy = self.reception_energy(self.number_of_packets[i])
                 if self.remaining_energy[i] < transmission_energy:
-                    self.remaining_energy[i] = 0 
+                    # self.remaining_energy[i] = 0 
                     continue  # Skip if the sensor does not have enough energy to transmit data to the next hop
+                
+                self.update_sensor_energies(i, transmission_energy)
+                # Update the metrics
+                self.total_energy_consumed += transmission_energy
+                self.total_packets_sent_by_sensors += self.number_of_packets[i]
+
                 if self.remaining_energy[action] < reception_energy:
-                    self.number_of_packets[i] = 0
-                    self.remaining_energy[action] = 0
+                    # self.remaining_energy[action] = 0
+                    self.packet_latency[i] = 0 # Reset the latency of the packet
+                    self.number_of_packets[i] = 0 # Reset the number of packets of the sensor i
                     continue  # Skip if the next hop does not have enough energy to receive data
 
-                self.update_sensor_energies(i, transmission_energy)  
                 self.update_sensor_energies(action, reception_energy) 
 
                 # Update the metrics
-                self.total_energy_consumed += transmission_energy + reception_energy
-                self.total_packets_sent_by_sensors += self.number_of_packets[i]
+                self.total_energy_consumed += reception_energy
                 self.packet_latency[action] += self.packet_latency[i] + latency_per_hop
                 self.packet_latency[i] = 0
 
@@ -210,7 +217,7 @@ class WSNRoutingEnv(gym.Env):
                 # Update the number of packets
                 self.number_of_packets[action] += self.number_of_packets[i]
                 
-            self.number_of_packets[i] = 0 # Reset the number of packets of the sensor i
+                self.number_of_packets[i] = 0 # Reset the number of packets of the sensor i
         
         # Integrate the mobility of the sensors
         # self.integrate_mobility() 
@@ -497,7 +504,7 @@ class WSNRoutingEnv(gym.Env):
         # Weighted sum of the rewards
         final_reward = sum([attention_weight * reward for attention_weight, reward in zip(attention_weights, rewards)])
 
-        return np.mean(final_reward)
+        return np.sum(final_reward)
 
 
     def integrate_mobility(self):

{gym_examples-3.0.765.dist-info → gym_examples-3.0.767.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=lI2A534ndKHi9RQYvkCYTX1c8O0mQr5rLLLQuRzu5Qw,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=QX9nNxbq87fRr0Dtfwe3aPtTmtiCJtNHixJLfycR1mo,26794
+gym_examples-3.0.767.dist-info/METADATA,sha256=KSj6oEzWvhbdwcJxk6M1V_OxbwExmam4LyXCHUbrWZ8,412
+gym_examples-3.0.767.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.767.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.767.dist-info/RECORD,,

gym_examples-3.0.765.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=lAcZyNHFpvqay_Ku_ZPsmDHzPBLETVDYUAJlyG-Et30,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=P0SHQeeyYeLUW6XxQTQ5fiDhsOr2IJVxGVjr8FzF0Eg,26461
-gym_examples-3.0.765.dist-info/METADATA,sha256=QZEeaeItoN1ALsY08KtHUnzCZoNOqlk9slZuSfnX3fw,412
-gym_examples-3.0.765.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.765.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.765.dist-info/RECORD,,