From aa9ae524c95a03e543e2faa443ece65570bbf580 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Fri, 19 Sep 2025 20:55:31 +0200 Subject: [PATCH] fix(reforcexy): fix access to _total_timesteps property MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index def7997..2370f75 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -2002,9 +2002,10 @@ class InfoMetricsCallback(TensorboardCallback): except Exception: pass - if not np.isclose(self._total_timesteps, 0.0): + total_timesteps = getattr(self.model, "_total_timesteps", None) + if total_timesteps is not None and not np.isclose(total_timesteps, 0.0): try: - progress_done = float(self.num_timesteps) / float(self._total_timesteps) + progress_done = float(self.num_timesteps) / float(total_timesteps) progress_done = ( 0.0 if progress_done < 0 -- 2.43.0