Corey Morris commited on
Commit
dd61816
1 Parent(s): a89ad93

changed to save and load in a directory

Browse files
Files changed (2) hide show
  1. save_for_regression.py +1 -1
  2. test_regression.py +1 -1
save_for_regression.py CHANGED
@@ -45,7 +45,7 @@ if __name__ == '__main__':
45
  last_commit = os.popen('git rev-parse HEAD').read().strip()
46
  print(last_commit)
47
  # save the current output to a file
48
- df_current.to_parquet(f'output_{last_commit}.parquet', index=True)
49
  print("Saved output to file")
50
  if untracked_changes:
51
  print("There are untracked changes")
 
45
  last_commit = os.popen('git rev-parse HEAD').read().strip()
46
  print(last_commit)
47
  # save the current output to a file
48
+ df_current.to_parquet(f'dataframe_history/output_{last_commit}.parquet', index=True)
49
  print("Saved output to file")
50
  if untracked_changes:
51
  print("There are untracked changes")
test_regression.py CHANGED
@@ -12,7 +12,7 @@ class TestRegression(unittest.TestCase):
12
  # load the reference dataframe
13
  last_commit = os.popen('git rev-parse HEAD').read().strip()
14
  print(last_commit)
15
- reference_file = f'output_{last_commit}.parquet'
16
  df_reference = pd.read_parquet(reference_file)
17
 
18
 
 
12
  # load the reference dataframe
13
  last_commit = os.popen('git rev-parse HEAD').read().strip()
14
  print(last_commit)
15
+ reference_file = f'dataframe_history/output_{last_commit}.parquet'
16
  df_reference = pd.read_parquet(reference_file)
17
 
18