From dd89d3b6c148b123c9f6c6e9f1dfab90b64c5e7a Mon Sep 17 00:00:00 2001 From: Edan Toledo Date: Wed, 16 Oct 2024 12:01:45 -0400 Subject: [PATCH] fix: add specific system condition for envpool installation --- .pre-commit-config.yaml | 2 +- requirements/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 0c460adc..0a597c22 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -1,4 +1,4 @@ -default_stages: [ "commit", "commit-msg", "push" ] +default_stages: [ "pre-commit", "commit-msg", "pre-push" ] default_language_version: python: python3 diff --git a/requirements/requirements.txt b/requirements/requirements.txt index 18c118b7..0b69801b 100644 --- a/requirements/requirements.txt +++ b/requirements/requirements.txt @@ -3,7 +3,7 @@ chex colorama craftax distrax @ git+https://github.com/google-deepmind/distrax # distrax release doesn't support jax > 0.4.13 -envpool +envpool; sys_platform == "linux" flashbax @ git+https://github.com/instadeepai/flashbax flax gymnasium