Python Script - batch size error