raw parameter set as default

casts argument un for_each_col as pandas series
UDST · Oct 11, 2020 · 0876a1d · 0876a1d
1 parent 7751997
commit 0876a1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/synthpop/ipu/ipu.py b/synthpop/ipu/ipu.py
@@ -25,7 +25,7 @@ def for_each_col(col):
         nz = col.to_numpy().nonzero()[0]
         return col.iloc[nz], nz
 
-    for (col_idx, (col, nz)) in df.apply(for_each_col, axis=0, raw=True).items():
+    for (col_idx, (col, nz)) in df.apply(for_each_col, axis=0, raw=False).items():
         yield (col_idx, col, nz)