From 5eeb051e4a34ca87eb7a5e0d0682562188b48953 Mon Sep 17 00:00:00 2001
From: Matthew Rocklin <mrocklin@gmail.com>
Date: Sat, 9 Mar 2019 23:09:10 -0800
Subject: [PATCH] flake8

---
 dask/dataframe/core.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dask/dataframe/core.py b/dask/dataframe/core.py
index d8057cf0f19..da51e09ce43 100644
--- a/dask/dataframe/core.py
+++ b/dask/dataframe/core.py
@@ -3812,7 +3812,7 @@ def map_partitions(func, *args, **kwargs):
     if transform_divisions and isinstance(dfs[0], Index) and len(dfs) == 1:
         try:
             divisions = func(
-                *[pd.Index(arg.divisions) if arg is dfs[0] else arg for arg in args],
+                *[pd.Index(a.divisions) if a is dfs[0] else a for a in args],
                 **kwargs
             )
             if isinstance(divisions, pd.Index):