FIX : avoid warning in test_time_gen.py