Minor reformatting in data_pipeline.py