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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. gym_examples/__init__.py +1 -1
  2. gym_examples/envs/wsn_env.py +4 -1
  3. {gym_examples-3.0.752.dist-info → gym_examples-3.0.753.dist-info}/METADATA +1 -1
  4. gym_examples-3.0.753.dist-info/RECORD +7 -0
  5. gym_examples-3.0.752.dist-info/RECORD +0 -7
  6. {gym_examples-3.0.752.dist-info → gym_examples-3.0.753.dist-info}/WHEEL +0 -0
  7. {gym_examples-3.0.752.dist-info → gym_examples-3.0.753.dist-info}/top_level.txt +0 -0
gym_examples/__init__.py CHANGED
@@ -5,4 +5,4 @@ register(
5
5
  entry_point="gym_examples.envs:WSNRoutingEnv",
6
6
  )
7
7
 
8
- __version__ = "3.0.752"
8
+ __version__ = "3.0.753"
gym_examples/envs/wsn_env.py CHANGED
@@ -48,7 +48,7 @@ class ScalarAttentionModel(nn.Module):
48
48
  # Apply GaussianAdaptiveAttention
49
49
  attention_output = self.ga_attention(x)
50
50
  # Map to scalar
51
- scalar_output = self.output_layer(attention_output).mean()
51
+ scalar_output = self.output_layer(attention_output)
52
52
  return scalar_output
53
53
 
54
54
  net = ScalarAttentionModel(input_dim)
@@ -470,8 +470,11 @@ class WSNRoutingEnv(gym.Env):
470
470
  '''
471
471
  Compute the attention-based rewards
472
472
  '''
473
+ print(f"Before torch tensor operation\nRewards type: {type(rewards)}, Rewards shape: {np.array(rewards).shape}")
473
474
  rewards = torch.tensor(rewards, dtype=torch.double)
475
+ print(f"After torch tensor operation\nRewards type: {type(rewards)}, Rewards shape: {rewards.shape}")
474
476
  rewards = rewards.unsqueeze(0) # Add batch dimension
477
+ print(f"After unsqueeze, rewards type: {type(rewards)}, rewards shape: {rewards.shape}")
475
478
  final_reward = net(rewards)
476
479
  return final_reward
477
480
 
{gym_examples-3.0.752.dist-info → gym_examples-3.0.753.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: gym-examples
3
- Version: 3.0.752
3
+ Version: 3.0.753
4
4
  Summary: A custom environment for multi-agent reinforcement learning focused on WSN routing.
5
5
  Home-page: https://github.com/gedji/CODES.git
6
6
  Author: Georges Djimefo
gym_examples-3.0.753.dist-info/RECORD ADDED
@@ -0,0 +1,7 @@
1
+ gym_examples/__init__.py,sha256=HPc6ReDuVSkSVN757l9_qqSk5jUFLgJf2quo1k9tnnw,166
2
+ gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
3
+ gym_examples/envs/wsn_env.py,sha256=DX1Ejs3qj417OAFgCi4Y2Bq7kFX8wUFp3flGeX0IZPI,27008
4
+ gym_examples-3.0.753.dist-info/METADATA,sha256=9kxvKD1m4MgAjh6PeOfoWA6DlEgIdMlnsvC_mttYQxU,412
5
+ gym_examples-3.0.753.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
6
+ gym_examples-3.0.753.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
7
+ gym_examples-3.0.753.dist-info/RECORD,,
gym_examples-3.0.752.dist-info/RECORD DELETED
@@ -1,7 +0,0 @@
1
- gym_examples/__init__.py,sha256=cSYjJ492gSXyqSOk2FqKhhWRXCx7ZzFuBbf1AuBd038,166
2
- gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
3
- gym_examples/envs/wsn_env.py,sha256=Cb6Og_FyZfyy1xrglch7CdUBsGvFT0QEpyX_cHCc8C4,26684
4
- gym_examples-3.0.752.dist-info/METADATA,sha256=_ZQEIPcDa066N80eabTuL1dFKRBgRgwa131wsKmcpxU,412
5
- gym_examples-3.0.752.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
6
- gym_examples-3.0.752.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
7
- gym_examples-3.0.752.dist-info/RECORD,,