Merge pull request #46 from dongchirua/master

- package version updated
- fix minor bug

Author: koulanurag

.github/workflows/python-package.yml

@@ -29,7 +29,7 @@ jobs:
     strategy:
       matrix:
         os: [macos-latest, ubuntu-latest]
-        python-version: [3.6, 3.7, 3.8]
+        python-version: [3.6, 3.7, 3.8, 3.9]
 
     steps:
     - uses: actions/checkout@v2

ma_gym/envs/switch/switch_one_corridor.py

@@ -26,6 +26,7 @@ def __init__(self, full_observable: bool = False, step_cost: float = 0, n_agents
         self._step_cost = step_cost
         self._total_episode_reward = None
         self._add_clock = clock
+        self._agent_dones = None
 
         self.action_space = MultiAgentActionSpace([spaces.Discrete(5) for _ in range(self.n_agents)])  # l,r,t,d,noop
 

ma_gym/envs/traffic_junction/traffic_junction.py

@@ -242,7 +242,7 @@ def get_agent_obs(self):
         agent_obs = []
         for agent_i in range(self.n_agents):
             pos = self.agent_pos[agent_i]
-            mask_view = np.zeros((*self._agent_view_mask, len(agent_no_mask_obs[0])))
+            mask_view = np.zeros((*self._agent_view_mask, len(agent_no_mask_obs[0])), dtype=np.float32)
             for row in range(max(0, pos[0] - 1), min(pos[0] + 1 + 1, self._grid_shape[0])):
                 for col in range(max(0, pos[1] - 1), min(pos[1] + 1 + 1, self._grid_shape[1])):
                     if PRE_IDS['agent'] in self._full_obs[row][col]:

setup.py

@@ -12,7 +12,7 @@
 extras['all'] = [item for group in extras.values() for item in group]
 
 setup(name='ma_gym',
-      version='0.0.10',
+      version='0.0.12',
       description='A collection of multi agent environments based on OpenAI gym.',
       long_description_content_type='text/markdown',
       long_description=open(path.join(path.abspath(path.dirname(__file__)), 'README.md'), encoding='utf-8').read(),
@@ -25,8 +25,8 @@
           'scipy>=1.3.0',
           'numpy>=1.16.4',
           'pyglet>=1.4.0,<=1.5.27',
-          'cloudpickle>=1.2.0,<1.7.0',
-          'gym==0.19.0',
+          'cloudpickle==2.0.0',
+          'gym>=0.19.0,<=0.20.0',
           'pillow>=7.2.0',
           'six>=1.16.0'
       ],
@@ -37,5 +37,6 @@
           'Programming Language :: Python :: 3.6',
           'Programming Language :: Python :: 3.7',
           'Programming Language :: Python :: 3.8',
+          'Programming Language :: Python :: 3.9',
       ],
       )