diff --git a/run_backtest.py b/run_backtest.py index 87e196c..3b33279 100644 --- a/run_backtest.py +++ b/run_backtest.py @@ -50,10 +50,12 @@ def run_backtest(start_date: datetime, end_date: datetime): start_date, end_date ) - # TODO: Think of a better way to handle this. - backtest_results.drop('Cumulative Profit', axis = 1, inplace = True) - print(backtest_results) - insert(backtest_results) + + if not backtest_results.empty: + # TODO: Think of a better way to handle this. + backtest_results.drop('Cumulative Profit', axis = 1, inplace = True) + print(backtest_results) + insert(backtest_results) if __name__ == '__main__': # TODO: Update backtest to eliminate the need for setting this to midnight.