gym-examples 3.0.135__py3-none-any.whl → 3.0.137__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "3.0.135"
+__version__ = "3.0.137"

gym_examples/envs/wsn_env.py

@@ -59,19 +59,11 @@ 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 = gym.spaces.Dict({
-            'remaining_energy': gym.spaces.Box(low=0, high=initial_energy, shape=(self.n_agents,), dtype=np.float32),
-            'consumption_energy': gym.spaces.Box(low=0, high=initial_energy, shape=(self.n_agents,), dtype=np.float32),
-            'sensor_positions': gym.spaces.Box(low=lower_bound, high=upper_bound, shape=(self.n_agents, 2), dtype=np.float32),
-            'number_of_packets': gym.spaces.Box(low=0, high=self.n_sensors * initial_number_of_packets + 1, shape=(self.n_agents,), dtype=int)
-        })
+        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))
-        self.action_space = gym.spaces.MultiDiscrete([self.n_sensors + 1] * self.n_agents)
+        self.action_space = Tuple(tuple([Discrete(self.n_sensors + 1)] * self.n_agents))
 
         self.reset()        
 
@@ -181,38 +173,21 @@ 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_observation_space(self):
-    #     return Dict({
-    #         'remaining_energy': Box(low=0, high=initial_energy, shape=(1,), dtype=np.float64),
-    #         'consumption_energy': Box(low=0, high=initial_energy, shape=(1,), dtype=np.float64),
-    #         'sensor_positions': Box(low=lower_bound, high=upper_bound, shape=(2,), dtype=np.float64),
-    #         'number_of_packets': Box(low=0, high=self.n_sensors * initial_number_of_packets + 1, shape=(1,), dtype=int)
-    #     })
-
     def _get_obs(self):
-        return {
-            'remaining_energy': np.array(self.remaining_energy),
-            'consumption_energy': np.array([initial_energy - e for e in self.remaining_energy]),
-            'sensor_positions': np.stack(self.sensor_positions, axis=0),
-            'number_of_packets': np.array(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_observation_space(self):
-        return gym.spaces.Dict({
-            'remaining_energy': gym.spaces.Box(low=0, high=initial_energy, shape=(len(self.remaining_energy),), dtype=np.float32),
-            'consumption_energy': gym.spaces.Box(low=0, high=initial_energy, shape=(len(self.remaining_energy),), dtype=np.float32),
-            'sensor_positions': gym.spaces.Box(low=lower_bound, high=upper_bound, shape=(len(self.sensor_positions), 2), dtype=np.float32),
-            'number_of_packets': gym.spaces.Box(low=0, high=self.n_sensors * initial_number_of_packets + 1, shape=(len(self.number_of_packets),), dtype=int)
+        return Dict({
+            'remaining_energy': Box(low=0, high=initial_energy, shape=(1,), dtype=np.float64),
+            'consumption_energy': Box(low=0, high=initial_energy, shape=(1,), dtype=np.float64),
+            'sensor_positions': Box(low=lower_bound, high=upper_bound, shape=(2,), dtype=np.float64),
+            'number_of_packets': Box(low=0, high=self.n_sensors * initial_number_of_packets + 1, shape=(1,), dtype=int)
         })
 
-
     def get_state(self):
         return self._get_obs()
     

{gym_examples-3.0.135.dist-info → gym_examples-3.0.137.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=EdR1SDwVV8SzSIJ4Gr54QXNHDqfRN9OQuvVu86vVSZ4,194
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=baMc1CiZz4NgtfUQ8LJW3057TCL9dPnuNEsreHMsSzs,21517
+gym_examples-3.0.137.dist-info/METADATA,sha256=ybEfreDxfH-l3TFQdMGYJ4KBxunwS4ALQkGqcROtcAg,412
+gym_examples-3.0.137.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.137.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.137.dist-info/RECORD,,

gym_examples-3.0.135.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=AtaFL7DdMIJNpqPaITYYoHuygF_KlV9TJ_6h7tHuVh0,194
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=HSTtatueYT5XWeAxyNhryTdoaGoB3n9h0zdRZ_K9Q84,23236
-gym_examples-3.0.135.dist-info/METADATA,sha256=WbDIlcgIXn1PoECyUQJVa783GuenQxz2bfaTDzaZSHA,412
-gym_examples-3.0.135.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.135.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.135.dist-info/RECORD,,