From 36a1d1e4ff651b259a5b0a56201a98c3844e1b0c Mon Sep 17 00:00:00 2001
From: John Aslanides <jaslanides@google.com>
Date: Wed, 30 Sep 2020 01:59:50 -0700
Subject: [PATCH] Bump version again (incorrect tag was used before) and use
 PyPI's rlax.

PiperOrigin-RevId: 334553236
Change-Id: I88eca16269fabf975dd0c8f7ccd8a8c2374b78c9
---
 bsuite/_metadata.py | 2 +-
 setup.py            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/bsuite/_metadata.py b/bsuite/_metadata.py
index 123d557..598cbcc 100644
--- a/bsuite/_metadata.py
+++ b/bsuite/_metadata.py
@@ -20,4 +20,4 @@
 a time when bsuite's dependencies may not have been installed yet.
 """
 
-__version__ = '0.3.3'
+__version__ = '0.3.4'
diff --git a/setup.py b/setup.py
index 0a48a57..032219a 100755
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,7 @@
     'jax',
     'jaxlib',
     'optax',
-    'rlax @ git+git://github.com/deepmind/rlax.git#egg=rlax',
+    'rlax',
     'tqdm',
 ]