refactor(reforcexy): refine typing

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 cbf6f1a594e1e320d6d54bcbcb5e381dbbf53de8..f4bee38d6caed0bd8f83e877cb52a65657102997 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -1926,7 +1926,7 @@ class InfoMetricsCallback(TensorboardCallback):
 
     @staticmethod
     def _build_train_freq(
-        train_freq: Optional[Union[int, Tuple[int], List[int]]],
+        train_freq: Optional[Union[TrainFreq, int, Tuple[int], List[int]]],
     ) -> Optional[int]:
         train_freq_val: Optional[int] = None
         if isinstance(train_freq, TrainFreq):