Apply formatting.

ikostrikov · Sep 28, 2021 · bc8030c · bc8030c
1 parent e9cb756
commit bc8030c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/jaxrl/networks/policies.py b/jaxrl/networks/policies.py
@@ -153,4 +153,5 @@ def sample_actions(
         observations: np.ndarray,
         temperature: float = 1.0,
         distribution: str = 'log_prob') -> Tuple[PRNGKey, jnp.ndarray]:
-    return _sample_actions(rng,actor_def,actor_params,observations,temperature,distribution)
+    return _sample_actions(rng, actor_def, actor_params, observations,
+                           temperature, distribution)
diff --git a/setup.py b/setup.py
@@ -9,8 +9,8 @@
 install_requires = [
     'numpy >= 1.20.2', 'scipy >= 1.6.0', 'absl-py >= 0.12.0',
     'gym[mujoco] >= 0.18.0', 'gdown >= 3.12.2', 'tqdm >= 4.60.0',
-    'flax >= 0.3.5', 'jax >= 0.2.21',
-    'ml_collections >= 0.1.0', 'optax >= 0.0.6', 'tensorboardX == 2.1',
+    'flax >= 0.3.5', 'jax >= 0.2.21', 'ml_collections >= 0.1.0',
+    'optax >= 0.0.6', 'tensorboardX == 2.1',
     'tensorflow-probability >= 0.13.0', 'imageio >= 2.9.0',
     'imageio-ffmpeg >= 0.4.3',
     'd4rl @ git+https://github.com/rail-berkeley/d4rl@master#egg=d4rl'