gym-examples 3.0.50__py3-none-any.whl → 3.0.51__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.50"
+__version__ = "3.0.51"

gym_examples/envs/wsn_env.py

@@ -10,7 +10,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 
 # Define the network parameters for the final reward function
-input_dim = 4  # length of the individual rewards vector
+# input_dim = 4  # length of the individual rewards vector
 output_dim = 1  # final reward
 
 Eelec = 50e-9  # energy consumption per bit in joules
@@ -121,8 +121,8 @@ class WSNRoutingEnv(gym.Env):
                 self.total_latency += self.packet_latency[i] + latency_per_hop
                 self.packet_latency[i] = 0
 
-                # rewards[i] = self.compute_individual_rewards(i, action)
-                rewards[i] = np.ones(input_dim)  # Reward for transmitting data to the base station
+                rewards[i] = self.compute_individual_rewards(i, action)
+                # rewards[i] = np.ones(input_dim)  # Reward for transmitting data to the base station
                 dones[i] = True       
             else:
                 distance = np.linalg.norm(self.sensor_positions[i] - self.sensor_positions[action])
@@ -363,8 +363,10 @@ class WSNRoutingEnv(gym.Env):
         '''
         Compute the attention-based rewards
         '''
+        input_dim = len(rewards)
         rewards = torch.tensor(rewards, dtype=torch.double)
-        net = Attention(len(rewards), output_dim)
+        print(f"input_dim_modified: {len(rewards)}")
+        net = Attention(input_dim, output_dim)
         net = net.double()  # Convert the weights to Double
         final_reward = net(rewards)
         return final_reward 

{gym_examples-3.0.50.dist-info → gym_examples-3.0.51.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=Mw543YXe6ddweCkbMiBJIu5taTJQMEm-gNIu3Omp8e8,193
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=DuacU7Dou0YVfINsYsR7NorKUgchUdz7Px5-rwlKFAM,20034
+gym_examples-3.0.51.dist-info/METADATA,sha256=vopRCS5_QmK3Gc2le27GqNMnovSG4lZtGoeWXFyyy1M,411
+gym_examples-3.0.51.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.51.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.51.dist-info/RECORD,,

gym_examples-3.0.50.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=5bGZE79MM87Lsi9G4pt9pzYriZjrdSe-u9MppmESRFY,193
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=S4HYOGOsORx63BvKGPwYhDlRnVITWU6cY1ZN2lZLzDQ,19947
-gym_examples-3.0.50.dist-info/METADATA,sha256=nUkf1NtLaCSfGMpgdwb7IzO_HIVUrbS5EfFsDFCzUtE,411
-gym_examples-3.0.50.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.50.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.50.dist-info/RECORD,,