Add back check for null columns

PSLmodels · Feb 27, 2021 · 6e9826b · 6e9826b
1 parent 814bcec
commit 6e9826b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scf/load.py b/scf/load.py
@@ -70,7 +70,8 @@ def load(
     :rtype: Union[pd.DataFrame, mdf.MicroDataFrame]
     """
     # Make cols a list if a single column is passed.
-    columns = mdf.listify(columns)
+    if columns is not None:
+        columns = mdf.listify(columns)
     # If years is a single year rather than a list, don't use a loop.
     if isinstance(years, int):
         res = load_single_scf(years, columns)