Skip to content

Commit

Permalink
Update pipeline.py
Browse files Browse the repository at this point in the history
import all state cleaners
  • Loading branch information
averyschoen authored Dec 5, 2023
1 parent 5c79dd6 commit a832404
Showing 1 changed file with 9 additions and 8 deletions.
17 changes: 9 additions & 8 deletions utils/pipeline.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,14 @@
import pandas as pd
from utils.arizona import ArizonaCleaner
from utils.michigan import MichiganCleaner
from utils.minnesota import MinnesotaCleaner
from utils.pennsylvania import PennsylvaniaCleaner

# import state cleaners here

# uncomment your state once it is added
state_cleaners = [
# ArizonaCleaner(),
# MichiganCleaner(),
# MinnesotaCleaner(),
# PennsylvaniaCleaner(),
ArizonaCleaner(),
MichiganCleaner(),
MinnesotaCleaner(),
PennsylvaniaCleaner(),
]

if __name__ == "__main__":
Expand All @@ -26,4 +27,4 @@

complete_individuals_table = pd.concat(single_state_individuals_tables)
complete_organizations_table = pd.concat(single_state_organizations_tables)
complete_transactions_table = pd.concat(single_state_transactions_tables)
# complete_transactions_table = pd.concat(single_state_transactions_tables)

0 comments on commit a832404

Please sign in to comment.