diff --git a/.gitignore b/.gitignore index cf0445b..e66f245 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,5 @@ stock-data/ readme.txt *.pickle +*.pkl *.html diff --git a/program.py b/program.py index 465bf00..636ab21 100644 --- a/program.py +++ b/program.py @@ -10,8 +10,8 @@ SCRIPT_LOC = os.path.join(os.path.dirname(os.path.realpath(__file__))) CIK_CSV_LOC = os.path.join(SCRIPT_LOC, 'cik.csv') -PICKLE_LOC_1 = os.path.join(SCRIPT_LOC, 'cik_dataframe.pickle') -PICKLE_LOC_2 = os.path.join(SCRIPT_LOC, 'sec_dataframe.pickle') +PICKLE_LOC_1 = os.path.join(SCRIPT_LOC, 'cik_dataframe.pkl') +PICKLE_LOC_2 = os.path.join(SCRIPT_LOC, 'sec_dataframe.pkl') HTML_LOC = os.path.join(SCRIPT_LOC, 'tables.html') HTML_COLUMNS = [