Fix warning in tools/data_samples.py