gym-examples 3.0.129__py3-none-any.whl → 3.0.130__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.129"
+__version__ = "3.0.130"

gym_examples/envs/wsn_env.py

@@ -172,12 +172,26 @@ class WSNRoutingEnv(gym.Env):
 
         return self._get_obs(), rewards, dones, {}
 
+    # def _get_obs(self):
+    #     return [{'remaining_energy': np.array([e]), 
+    #              'consumption_energy': np.array([initial_energy - e]),
+    #              'sensor_positions': p,
+    #              'number_of_packets': np.array([d])
+    #             } for e, p, d in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)]
+
     def _get_obs(self):
-        return [{'remaining_energy': np.array([e]), 
-                 'consumption_energy': np.array([initial_energy - e]),
-                 'sensor_positions': p,
-                 'number_of_packets': np.array([d])
-                } for e, p, d in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)]
+        remaining_energy = np.array([e for e, _, _, _ in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)])
+        consumption_energy = np.array([initial_energy - e for e, _, _, _ in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)])
+        sensor_positions = np.stack(self.sensor_positions, axis=0)
+        number_of_packets = np.array([d for _, _, _, d in zip(self.remaining_energy, self.sensor_positions, self.number_of_packets)])
+
+        return {
+            'remaining_energy': remaining_energy,
+            'consumption_energy': consumption_energy,
+            'sensor_positions': sensor_positions,
+            'number_of_packets': number_of_packets
+        }
+
 
     def _get_observation_space(self):
         return Dict({

{gym_examples-3.0.129.dist-info → gym_examples-3.0.130.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=ephqHG1V21jR1hU69gk7ByNV2lo-xcz__sSSxL0TRiQ,194
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=jPME2gvkDL7BfMqYgjpAXna-mTAv3_81HlPs-U5FnTM,22286
+gym_examples-3.0.130.dist-info/METADATA,sha256=28_KBJf-MOIyD0HgbmzMvu1UlAcH5s-o0az8Qd1pN1Y,412
+gym_examples-3.0.130.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.130.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.130.dist-info/RECORD,,

gym_examples-3.0.129.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=f5RfZkZxk0_sPLzl68-buUGDEr0bw8Odn_OhBrqtJMI,194
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=SPuZ6Eezn7LY38r7eGtMAZw6SfbLnBSdbGyLNaSA0D0,21515
-gym_examples-3.0.129.dist-info/METADATA,sha256=kKkDZbChu1-rph3VjmRNqfLqMhfRP7jQkn-OpZeKxGQ,412
-gym_examples-3.0.129.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.129.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.129.dist-info/RECORD,,