gym-examples 2.0.1__py3-none-any.whl → 2.0.3__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "2.0.1"
+__version__ = "2.0.3"

gym_examples/envs/wsn_env.py

@@ -47,7 +47,7 @@ net = net.double()  # Convert the weights to Double
 
 
 class WSNRoutingEnv(gym.Env):
-    def __init__(self, n_sensors = 20, coverage_radius=50):
+    def __init__(self, n_sensors = 10, coverage_radius=50):
 
         super(WSNRoutingEnv, self).__init__()
 
@@ -76,62 +76,100 @@ class WSNRoutingEnv(gym.Env):
         self.consumption_energy = np.zeros(self.n_sensors)
         self.number_of_packets = np.ones(self.n_sensors, dtype=int)  # Number of packets to transmit
         self.episode_count += 1
-    
+        self.step_count = 0
+        print("\n=====================================")
+        print(f"Episode {self.episode_count} started") 
+        print(f"Sensor positions: {self.sensor_positions}")
+        print(f"Distance to base station: {self.distance_to_base}")
+        print(f"Remaining energy: {self.remaining_energy}")
+        print(f"Number of packets: {self.number_of_packets}")
+        print(f"Consumption energy: {self.consumption_energy}")
+        print("=====================================\n")
         return self._get_obs()
 
     def step(self, actions):
         rewards = [0] * self.n_sensors
         dones = [False] * self.n_sensors
+        print("\n=====================================")
+        print(f"Step {self.step_count + 1} started")
+        print(f"Sensor positions: {self.sensor_positions}")
+        print(f"Distance to base station: {self.distance_to_base}")
+        print(f"Remaining energy: {self.remaining_energy}")
+        print(f"Number of packets: {self.number_of_packets}")
+        print(f"Consumption energy: {self.consumption_energy}")
+        print("=====================================\n")
+        self.step_count += 1
         for i, action in enumerate(actions):
             
             if action not in range(self.n_sensors + 1):
+                print(f"Invalid action: {action} for sensor {i}!")
                 raise ValueError("Invalid action!")
             
             if i >= self.n_sensors:
+                print(f"Invalid sensor i: {i}!")
                 continue  # Skip if the number of actions is greater than the number of sensors
 
             if self.remaining_energy[i] <= 0 or self.number_of_packets[i] <= 0:
+                print(f"Sensor {i} has no energy left or no packets to transmit!")
+                print(f"Remaining energy: {self.remaining_energy[i]}")
+                print(f"Number of packets: {self.number_of_packets[i]}")
                 continue  # Skip if sensor has no energy left or no packets to transmit
             
             if (action == i):
+                print(f"Sensor {i} tries to transmit data to itself!")
                 continue  # Skip if sensor tries to transmit data to itself
 
             neighbors_i = self.eligible_receivers(i)
+            print(f"Sensor {i} eligible receivers: {neighbors_i}")
             keys_neighbors_i = list(neighbors_i.keys())
+            print(f"Sensor {i} keys of eligible receivers: {keys_neighbors_i}")
             if len(neighbors_i) == 0 or action not in keys_neighbors_i:
+                print(f"Sensor {i} has no eligible receivers or action {action} is not in the list of eligible receivers!")
                 continue
 
             remaining_energy_before = copy.deepcopy(self.remaining_energy)
+            print(f"Remaining energy before transmission: {remaining_energy_before}")
             if action == self.n_sensors:
                 rewards[i] = self.compute_individual_rewards(i, action, neighbors_i, remaining_energy_before)
+                print(f"Sensor {i} transmits data to the base station")
+                print(f"Sensor {i} rewards: {rewards[i]}")
                 dones[i] = True
                 # Calculate the energy consumption and remaining for transmitting data to the base station
-                self.update_sensor_energies(i, neighbors_i[action]['transmission_energy'])        
+                self.update_sensor_energies(i, neighbors_i[action]['transmission_energy'])    
+                print(f"Remaining energy after transmission of sensor {i} to base station: {self.remaining_energy}")
+                print(f"Consumption energy after transmission of sensor {i} to base station: {self.consumption_energy}")    
             else:
                 self.update_sensor_energies(i, neighbors_i[action]['transmission_energy'])  
-                self.update_sensor_energies(action, neighbors_i[action]['reception_energy'])              
+                self.update_sensor_energies(action, neighbors_i[action]['reception_energy'])   
+                print(f"Remaining energy after transmission of sensor {i} to sensor {action}: {self.remaining_energy}")
+                print(f"Consumption energy after transmission of sensor {i} to sensor {action}: {self.consumption_energy}")           
                 # Update the number of packets of the sensor action
                 self.number_of_packets[action] += self.number_of_packets[i]
-                
+                print(f"Number of packets of sensor {action}: {self.number_of_packets[action]}")
                 self.distance_to_base[action] = np.linalg.norm(self.sensor_positions[action] - base_station_position)
-                
+                print(f"Distance to base station of sensor {action}: {self.distance_to_base[action]}")
                 # Compute individual rewards
                 rewards[i] = self.compute_individual_rewards(i, action, neighbors_i, remaining_energy_before)
-
+                print(f"Sensor {i} rewards: {rewards[i]}")
             self.number_of_packets[i] = 0 # Reset the number of packets of the sensor i
-            
+            print(f"Number of packets of sensor {i}: {self.number_of_packets[i]}")
             # Calculate final reward
             # rewards_individual = torch.tensor(rewards[i], dtype=torch.double)                
             # final_reward = net(rewards_individual)
             final_reward = sum(rewards[i])
             rewards[i] = final_reward
+            print(f"Final reward of sensor {i}: {rewards[i]}")
 
         for i in range(self.n_sensors):
             if (self.remaining_energy[i] <= 0) or (self.number_of_packets[i] <= 0):
+                print(f"Sensor {i} has no energy left or no packets to transmit!")
+                print(f"Remaining energy: {self.remaining_energy[i]}")
+                print(f"Number of packets: {self.number_of_packets[i]}")
                 dones[i] = True
 
         # Integrate the mobility of the sensors
         self.integrate_mobility() 
+        print(f"Sensor positions after mobility: {self.sensor_positions}")
 
         return self._get_obs(), rewards, dones, {}
 

{gym_examples-2.0.1.dist-info → gym_examples-2.0.3.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=NDVMqs-BmbAaaVEkGhg9t4AoaV_Jk0MRnorBIpJjRdA,192
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=b1wYD9zifgnfmVYmuwMtCxfNdvwQbVcQHzYLTFi5bWo,18304
+gym_examples-2.0.3.dist-info/METADATA,sha256=heuSoilS5UuI7OqwKDBgQmGC6h14FOUmHG2kICOSXuk,410
+gym_examples-2.0.3.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-2.0.3.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-2.0.3.dist-info/RECORD,,

gym_examples-2.0.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=NKjYttm8WTS4IxmCab4MDe_UPTMTpqsQgfa3sr_-XT4,192
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=98imEkP8AZQ-PVrS8T5S-E9cKOoedPwh-MwIMzZHq4o,15206
-gym_examples-2.0.1.dist-info/METADATA,sha256=ilp5Oc4FYsyNhznFdqsRmHjINn4LJ7D15xcKOEf8mtI,410
-gym_examples-2.0.1.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-2.0.1.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-2.0.1.dist-info/RECORD,,