Merge pull request #436 from andersonfrailey/sortcolumns

Sort PUF columns
PSLmodels · Sep 15, 2023 · bab4bf8 · bab4bf8
2 parents cb3465f + 697761c
commit bab4bf8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/taxdata/puf/finalprep.py b/taxdata/puf/finalprep.py
@@ -74,6 +74,9 @@ def finalprep(data):
     # - Rename 'filer' to 'data_source'
     data = data.rename(columns={"filer": "data_source"})
 
+    # - Sort columns to ensure every PUF is the same
+    data.sort_index(axis=1, inplace=True)
+
     return data