fix(reforcexy): fix access to _total_timesteps property

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index def79973b9e788d480b87c432d0831de3cb542a7..2370f75f13711f8ed37b0a4bc517d4ec87b116e6 100644 (file)
--- 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