removed unused imports in stage_data.py