gym-examples 2.0.7__py3-none-any.whl → 2.0.9__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "2.0.7"
+__version__ = "2.0.9"

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 = 3, coverage_radius=50):
+    def __init__(self, n_sensors = 30, coverage_radius=50):
 
         super(WSNRoutingEnv, self).__init__()
 
@@ -76,101 +76,57 @@ 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):
-        print(f"Actions: {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'])    
-                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}")    
+                self.update_sensor_energies(i, neighbors_i[action]['transmission_energy'])        
             else:
                 self.update_sensor_energies(i, neighbors_i[action]['transmission_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.7.dist-info → gym_examples-2.0.9.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=gqSyx1EEuQeAWNhqLrgfA_OQ8AWGNJ0ZHREx2Wvr6kk,192
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=Iq3O6LUZnOwIry3rZRcSkfe0jLfKJnyv5ZL18aDwAvQ,15137
+gym_examples-2.0.9.dist-info/METADATA,sha256=xjdl49NkTN3aMG215C5vqaflQgwwesIAQdUw_xBjRiE,410
+gym_examples-2.0.9.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-2.0.9.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-2.0.9.dist-info/RECORD,,

gym_examples-2.0.7.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=S24PMK01kJnGXPbYfpKMTHxTUI-SziAjOgRu5W-S7YY,192
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=_XVGI1ZWnUr0vhgBqV1HdyJzMoilFra_vMM27CnQceE,18341
-gym_examples-2.0.7.dist-info/METADATA,sha256=IcY8_pntZCRrWmjJN8A8WMRGut-0zVXfpD4V01IPpfQ,410
-gym_examples-2.0.7.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-2.0.7.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-2.0.7.dist-info/RECORD,,