From 1fdd805f484412e7ea31ddd04d85309e7f6d524a Mon Sep 17 00:00:00 2001
From: Kevin Zakka <zakka@google.com>
Date: Tue, 18 Feb 2025 11:12:07 -0800
Subject: [PATCH] Skip training if play_only is True. Fixes #58.

PiperOrigin-RevId: 728284290
Change-Id: I1ff15d308005512cd314b6ab145a7f09cfc0c0fc
---
 learning/train_jax_ppo.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/learning/train_jax_ppo.py b/learning/train_jax_ppo.py
index 74208f9..0e04191 100644
--- a/learning/train_jax_ppo.py
+++ b/learning/train_jax_ppo.py
@@ -163,6 +163,8 @@ def main(argv):
 
   if _NUM_TIMESTEPS.present:
     ppo_params.num_timesteps = _NUM_TIMESTEPS.value
+  if _PLAY_ONLY.present:
+    ppo_params.num_timesteps = 0
   if _NUM_EVALS.present:
     ppo_params.num_evals = _NUM_EVALS.value
   if _REWARD_SCALING.present: