gym-examples 3.0.753__py3-none-any.whl → 3.0.754__py3-none-any.whl

gym_examples/__init__.py

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

gym_examples/envs/wsn_env.py

@@ -470,13 +470,14 @@ class WSNRoutingEnv(gym.Env):
         '''
         Compute the attention-based rewards
         '''
-        print(f"Before torch tensor operation\nRewards type: {type(rewards)}, Rewards shape: {np.array(rewards).shape}")
-        rewards = torch.tensor(rewards, dtype=torch.double)
-        print(f"After torch tensor operation\nRewards type: {type(rewards)}, Rewards shape: {rewards.shape}")
-        rewards = rewards.unsqueeze(0)  # Add batch dimension
-        print(f"After unsqueeze, rewards type: {type(rewards)}, rewards shape: {rewards.shape}")
-        final_reward = net(rewards)
-        return final_reward 
+        final_reward = []
+        for i in range(len(rewards[0])):
+            rewards_i = [reward[i] for reward in rewards]
+            rewards_i = torch.tensor(rewards_i, dtype=torch.double)
+            rewards_i = rewards_i.unsqueeze(0) # Add batch dimension
+            final_reward.append(net(rewards_i))
+        
+        return final_reward.mean().item() 
 
 
     # def compute_attention_reward(self, rewards):

{gym_examples-3.0.753.dist-info → gym_examples-3.0.754.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=AfgfcIhRLQ13gj-B6lbC4iTEtYaTWCO14JRFDumF1dA,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=I2px13TJyNn93cCQkO22hI5WuIJo_HT3v0ZfBqk7_0k,26856
+gym_examples-3.0.754.dist-info/METADATA,sha256=OrruAdPUHtFXCbUzZNRWkDB8MHpR2zMNqpGNY5ZFHOw,412
+gym_examples-3.0.754.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.754.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.754.dist-info/RECORD,,

gym_examples-3.0.753.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=HPc6ReDuVSkSVN757l9_qqSk5jUFLgJf2quo1k9tnnw,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=DX1Ejs3qj417OAFgCi4Y2Bq7kFX8wUFp3flGeX0IZPI,27008
-gym_examples-3.0.753.dist-info/METADATA,sha256=9kxvKD1m4MgAjh6PeOfoWA6DlEgIdMlnsvC_mttYQxU,412
-gym_examples-3.0.753.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.753.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.753.dist-info/RECORD,,