diff --git a/clrs/_src/samplers.py b/clrs/_src/samplers.py index 93351cc6..0b8e301a 100644 --- a/clrs/_src/samplers.py +++ b/clrs/_src/samplers.py @@ -106,7 +106,9 @@ def __init__( self._track_max_steps = track_max_steps if num_samples < 0: - logging.warning('Sampling dataset on-the-fly, unlimited samples.') + logging.log_first_n( + logging.WARNING, 'Sampling dataset on-the-fly, unlimited samples.', 1 + ) if track_max_steps: # Get an initial estimate of max hint length self.max_steps = -1