Sort PUF columns

PSLmodels · Sep 13, 2023 · 697761c · 697761c
1 parent cb3465f
commit 697761c
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