gym-examples 3.0.379__py3-none-any.whl → 3.0.381__py3-none-any.whl

gym_examples/__init__.py

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

gym_examples/envs/wsn_env.py

@@ -59,7 +59,8 @@ class Attention(nn.Module):
         # Step 1: Ensure input is 2D by adding a batch dimension if necessary
         if x.dim() == 1:
             x = x.unsqueeze(0)  # Shape: [1, input_dim]
-        x = F.relu(self.linear1(x))  # Pass the input through a linear layer and a ReLU activation function        
+        # x = F.relu(self.linear1(x))  # Pass the input through a linear layer and a ReLU activation function 
+        x = F.Softplus(self.linear1(x))  # Pass the input through a linear layer and a ReLU activation function       
         attention_weights = F.softmax(x, dim=1)  # Apply the softmax function to get the attention weights. Initially F.softmax(x, dim=0) 
         x = attention_weights * x  # Multiply the input by the attention weights
         x = self.linear2(x)  # Pass the result through another linear layer
@@ -196,7 +197,6 @@ class WSNRoutingEnv(gym.Env):
                 self.packet_latency[i] = 0
 
                 rewards[i] = [max_reward] * input_dim # Reward for transmitting data to the base station
-                print(f"Sensor {i} transmitted data to the base station with modified reward: {self.compute_attention_rewards(rewards[i])}")
                 dones[i] = True
             else:
                 distance = np.linalg.norm(self.sensor_positions[i] - self.sensor_positions[action])
@@ -223,11 +223,7 @@ class WSNRoutingEnv(gym.Env):
                 self.packet_latency[i] = 0
 
                 rewards[i] = self.compute_individual_rewards(i, action)
-                if self.number_of_steps > 27:
-                    raise Error("Stop here")
-                else:
-                    print(f"Sensor {i} transmitted data to sensor {action} with modified reward: {self.compute_attention_rewards(rewards[i])}") 
-
+                
                 # Update the number of packets
                 self.number_of_packets[action] += self.number_of_packets[i]
                 

{gym_examples-3.0.379.dist-info → gym_examples-3.0.381.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=-xF9v1R8lVyLNVmg8tKb1ZoPyow-1QIZ84QtzXD78rU,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=kbGt7hOHZIGso4NIyd0cm8TSbFx0RazDsXN8lAD5c6U,26990
+gym_examples-3.0.381.dist-info/METADATA,sha256=snKPXRfg3uUbz2zqWT1c6-GBVPPh7VDOomY2laV1HRs,412
+gym_examples-3.0.381.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.381.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.381.dist-info/RECORD,,

gym_examples-3.0.379.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=-ok5iR5eVxAwucS7NCfeCQPIdhS-25avnp1oxRQR1qg,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=3q-Xt1_1zZHatLkKYvkJUPYpHYoysZtolvMODkMKYHg,27263
-gym_examples-3.0.379.dist-info/METADATA,sha256=jgNbjP_koNOXpHUe51r4Xwi0F_nwC9-9WApQF-D4Pv4,412
-gym_examples-3.0.379.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.379.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.379.dist-info/RECORD,,