gym-examples 3.0.750__py3-none-any.whl → 3.0.752__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 +2 -1
  3. {gym_examples-3.0.750.dist-info → gym_examples-3.0.752.dist-info}/METADATA +1 -1
  4. gym_examples-3.0.752.dist-info/RECORD +7 -0
  5. gym_examples-3.0.750.dist-info/RECORD +0 -7
  6. {gym_examples-3.0.750.dist-info → gym_examples-3.0.752.dist-info}/WHEEL +0 -0
  7. {gym_examples-3.0.750.dist-info → gym_examples-3.0.752.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.750"
8
+ __version__ = "3.0.752"
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)
51
+ scalar_output = self.output_layer(attention_output).mean()
52
52
  return scalar_output
53
53
 
54
54
  net = ScalarAttentionModel(input_dim)
@@ -228,6 +228,7 @@ class WSNRoutingEnv(gym.Env):
228
228
  # rewards = [reward.item() if isinstance(reward, torch.Tensor) else reward for reward in rewards] # Convert the reward to a float
229
229
 
230
230
  rewards = self.compute_attention_rewards(rewards)
231
+ print(f"Rewards: {rewards}")
231
232
  # rewards = np.mean(rewards)
232
233
 
233
234
  for i in range(self.n_sensors):
{gym_examples-3.0.750.dist-info → gym_examples-3.0.752.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: gym-examples
3
- Version: 3.0.750
3
+ Version: 3.0.752
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.752.dist-info/RECORD ADDED
@@ -0,0 +1,7 @@
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,,
gym_examples-3.0.750.dist-info/RECORD DELETED
@@ -1,7 +0,0 @@
1
- gym_examples/__init__.py,sha256=qaPZA7Q93uDypjo58Stoco6cD-o4R03zzeYSCAE2JhU,166
2
- gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
3
- gym_examples/envs/wsn_env.py,sha256=8ne-Dm4xfoY-AYDyzOiZ1RjDtG6RjuWrXiiFQSix8Ac,26639
4
- gym_examples-3.0.750.dist-info/METADATA,sha256=LdlfRyDsFXXtHKkA1m-f2Gmb0_wXxb2dQPwpYuVD0OM,412
5
- gym_examples-3.0.750.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
6
- gym_examples-3.0.750.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
7
- gym_examples-3.0.750.dist-info/RECORD,,