gym-examples 3.0.126__py3-none-any.whl → 3.0.128__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.126"
+__version__ = "3.0.128"

gym_examples/envs/wsn_env.py

@@ -59,11 +59,34 @@ class WSNRoutingEnv(gym.Env):
         # Initialize the position of the sensors randomly
 
         # Define observation space
-        self.observation_space = Tuple(
-            tuple([self._get_observation_space() for _ in range(self.n_sensors)])
+        # self.observation_space = Tuple(
+        #     tuple([self._get_observation_space() for _ in range(self.n_sensors)])
+        # )
+        # self.action_space = Tuple(tuple([Discrete(self.n_sensors + 1)] * self.n_agents))
+
+        # Define the flattened observation space
+        single_sensor_obs_space = self._get_observation_space()
+        obs_low = np.concatenate([
+            single_sensor_obs_space['remaining_energy'].low,
+            single_sensor_obs_space['consumption_energy'].low,
+            single_sensor_obs_space['sensor_positions'].low,
+            single_sensor_obs_space['number_of_packets'].low
+        ])
+        obs_high = np.concatenate([
+            single_sensor_obs_space['remaining_energy'].high,
+            single_sensor_obs_space['consumption_energy'].high,
+            single_sensor_obs_space['sensor_positions'].high,
+            single_sensor_obs_space['number_of_packets'].high
+        ])
+        self.observation_space = Box(
+            low=np.tile(obs_low, self.n_sensors),
+            high=np.tile(obs_high, self.n_sensors),
+            dtype=np.float32
         )
-        self.action_space = Tuple(tuple([Discrete(self.n_sensors + 1)] * self.n_agents))
-                
+
+        # Define the flattened action space
+        self.action_space = MultiDiscrete([self.n_sensors + 1] * self.n_agents)
+
         self.reset()        
 
     def reset(self):                
@@ -172,13 +195,28 @@ class WSNRoutingEnv(gym.Env):
 
         return self._get_obs(), rewards, dones, {}
 
-    def _get_obs(self):
+    # 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)]
 
-        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):
+        # Structured observation
+        structured_obs = [{'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)]
+
+        # Flattened observation for compatibility with Stable Baselines3
+        flat_obs = np.concatenate([
+            np.concatenate([obs['remaining_energy'], obs['consumption_energy'], obs['sensor_positions'], obs['number_of_packets']])
+            for obs in structured_obs
+        ])
+        return flat_obs
 
     def _get_observation_space(self):
         return Dict({

{gym_examples-3.0.126.dist-info → gym_examples-3.0.128.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=HtlSQQsIPVhRkM90RZRf4j2KCKHhI4jQzcBnLzxNT1U,194
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=cPfhzLLoAMoBonuR7PdgS-OjU9e1AbaJlgRo4AxHTqY,23329
+gym_examples-3.0.128.dist-info/METADATA,sha256=CCcUSZp8Tyavj_Gxy9eWmZH1XJgvrMbItHuVR6BlRXM,412
+gym_examples-3.0.128.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.128.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.128.dist-info/RECORD,,

gym_examples-3.0.126.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=rl8PAXoiDtLF7ZyDztNrX_Fr0KG9KzKHWiQ7FSiYF3s,194
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=2SL0mSy2hAjNpiV0PxkCyO49HBr-5an8zapgnNX6EuA,21533
-gym_examples-3.0.126.dist-info/METADATA,sha256=L4gO7s8aSNejF1RTBgw8drEmymu3Ysm8hsmVkm3Guos,412
-gym_examples-3.0.126.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.126.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.126.dist-info/RECORD,,