diff options
author | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2022-12-06 13:06:57 -0800 |
---|---|---|
committer | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2022-12-06 13:06:57 -0800 |
commit | f6f0fc29ef5a0a0bf060654ee7bf77dac37020f7 (patch) | |
tree | 64376469afadbfad97e5e80cf8920ae16185eb9f | |
parent | f9621beecb23f94f31c7028a4d805f977e3e8a68 (diff) |
make expiration flag work
-rw-r--r-- | openbb_terminal/stocks/options/tradier_model.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/openbb_terminal/stocks/options/tradier_model.py b/openbb_terminal/stocks/options/tradier_model.py index 6cf4322fbc0..330a8d33bea 100644 --- a/openbb_terminal/stocks/options/tradier_model.py +++ b/openbb_terminal/stocks/options/tradier_model.py @@ -146,7 +146,7 @@ def get_full_option_chain(symbol: str) -> pd.DataFrame: for expiry in expirations: options_dfs.append(get_option_chains(symbol, expiry)) - + options_df = pd.concat(options_dfs) options_df.set_index(keys="symbol", inplace=True) @@ -211,7 +211,10 @@ def get_full_option_chain(symbol: str) -> pd.DataFrame: ] options = pd.DataFrame(options_df, columns=options_columns) - + options = options.reset_index() + options.drop(labels = ['Ticker'], inplace = True, axis = 1) + options.rename(columns = {'Expiration': 'expiration'}, inplace = True) + return options |