From b0b9da1a5466fb953182db4b9ef7947197e4043a Mon Sep 17 00:00:00 2001
From: Theresa Eimer <t.eimer@ai.uni-hannover.de>
Date: Fri, 9 Feb 2024 11:02:56 +0100
Subject: [PATCH] feat: readme badges + even more specific brax version

---
 README.md | 5 +++++
 setup.py  | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 2a5ed021..c379ec96 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,11 @@
 <img align="left" width="80" src="./docs/source/figures/CARL_logo.png" alt="CARL">
 
 # – The Benchmark Library
+[![PyPI Version](https://img.shields.io/pypi/v/carl-bench.svg)](https://pypi.python.org/pypi/carl-bench)
+[![Test](https://github.com/automl/carl/actions/workflows/tests.yaml/badge.svg)](https://github.com/automl/carl/actions/workflows/tests.yaml)
+[![Doc Status](https://github.com/automl/carl/actions/workflows/docs.yaml/badge.svg)](https://github.com/automl/carl/actions/workflows/docs.yaml)
+[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+
 CARL (context adaptive RL) provides highly configurable contextual extensions
 to several well-known RL environments.
 It's designed to test your agent's generalization capabilities
diff --git a/setup.py b/setup.py
index d6ca10bd..1a84d934 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ def read_file(filepath: str) -> str:
         "gymnasium[box2d]>=0.27.1",
     ],
     "brax": [
-        "brax==0.9",
+        "brax==0.9.1",
         "protobuf>=3.17.3",
     ],
     "dm_control": [