From b782f33732eaef2f743be2356218a8a674e2170a Mon Sep 17 00:00:00 2001 From: Clemens Winter Date: Mon, 26 Feb 2024 08:32:50 -0800 Subject: [PATCH 1/2] fix missing case for GlobalCategoricalActionSpace? --- entity_gym/env/vec_env.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/entity_gym/env/vec_env.py b/entity_gym/env/vec_env.py index b03d0ca..eb9c528 100644 --- a/entity_gym/env/vec_env.py +++ b/entity_gym/env/vec_env.py @@ -347,7 +347,9 @@ def batch_obs( for atype, space in action_space.items(): if atype not in o.actions: if atype in action_masks: - if isinstance(space, CategoricalActionSpace): + if isinstance(space, CategoricalActionSpace) or isinstance( + space, GlobalCategoricalActionSpace + ): vec_action = action_masks[atype] assert isinstance(vec_action, VecCategoricalActionMask) vec_action.actors.push(np.zeros((0, 1), dtype=np.int64)) From 5ef6b4f0e59fa8ae4726091b215390d94c535bf3 Mon Sep 17 00:00:00 2001 From: Clemens Winter Date: Wed, 28 Feb 2024 08:52:14 -0800 Subject: [PATCH 2/2] Fix ci build? --- .github/workflows/checks.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/checks.yaml b/.github/workflows/checks.yaml index f3bdad4..78655dd 100644 --- a/.github/workflows/checks.yaml +++ b/.github/workflows/checks.yaml @@ -49,10 +49,10 @@ jobs: #---------------------------------------------- - name: Install dependencies if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' - run: poetry run pip install torch + run: pip install torch==1.12.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Install dependencies if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' - run: poetry run pip install torch-scatter + run: pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cpu.html #---------------------------------------------- # install your root project, if required #---------------------------------------------- @@ -109,10 +109,10 @@ jobs: #---------------------------------------------- - name: Install dependencies if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' - run: poetry run pip install torch + run: pip install torch==1.12.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Install dependencies if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' - run: poetry run pip install torch-scatter + run: pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cpu.html #---------------------------------------------- # install your root project, if required #----------------------------------------------