gym-examples 3.0.348__py3-none-any.whl → 3.0.350__py3-none-any.whl

gym_examples/__init__.py

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

gym_examples/envs/wsn_env.py

@@ -38,11 +38,10 @@ class Attention(nn.Module):
         self.linear2 = nn.Linear(64, output_dim)  # Define the second linear layer. It takes input of size '64' and outputs size 'output_dim'
 
     def forward(self, x):
-        x = F.relu(self.linear1(x))  # Pass the input through a linear layer and a ReLU activation function
-        print("==============================")
-        print(f"x.dim: {x.dim()}")
-        print("==============================")
-        raise ValueError("Stop here")
+        # 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        
         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

{gym_examples-3.0.348.dist-info → gym_examples-3.0.350.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=_bZArYSz7CqKCjGT21LpKQnWJ_-UBzNYtqQV9Lqzj64,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=Dv9LjdTJHfZ3c4NSIFNq9DePV4hFOkJ7IM9eCAkCpBE,25356
+gym_examples-3.0.350.dist-info/METADATA,sha256=9fufmXOzIAwbHnlwplD7NShDyjTFxG90cwVAq_jm4S8,412
+gym_examples-3.0.350.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.350.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.350.dist-info/RECORD,,

gym_examples-3.0.348.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=wF4URbgfzFU0k9BzXDG3rlWdNZK51fq3BjZVX24gw4g,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=K6GSlHNrO4JGYQXXX5VH2Avs2deW2RHc54UTkOWWR4Y,25359
-gym_examples-3.0.348.dist-info/METADATA,sha256=r5kFwvHGCVq_LU7ML-QNM5AuHuJdLxLEI83Bv3IA6Xk,412
-gym_examples-3.0.348.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.348.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.348.dist-info/RECORD,,