From 1d73e7c192bf2e475e3eb91242fca14563e97ab0 Mon Sep 17 00:00:00 2001 From: James Maslek Date: Fri, 26 Aug 2022 06:30:42 -0400 Subject: [PATCH] Fix economy/fred start date (#2421) * limit was cropping data * pylint * boop * fix limit --- data_sources_default.json | 2 +- openbb_terminal/economy/fred_model.py | 7 +------ openbb_terminal/economy/fred_view.py | 4 ++-- 3 files changed, 4 insertions(+), 9 deletions(-) diff --git a/data_sources_default.json b/data_sources_default.json index 133f9dbc0d5..af5b217b6e5 100644 --- a/data_sources_default.json +++ b/data_sources_default.json @@ -210,4 +210,4 @@ "polygon" ] } -} +} \ No newline at end of file diff --git a/openbb_terminal/economy/fred_model.py b/openbb_terminal/economy/fred_model.py index ebd20f86fcb..b27f0a187d2 100644 --- a/openbb_terminal/economy/fred_model.py +++ b/openbb_terminal/economy/fred_model.py @@ -186,7 +186,7 @@ def get_series_data( @log_start_end(log=logger) def get_aggregated_series_data( - series_ids: List[str], start_date: str = None, end_date: str = None, limit: int = -1 + series_ids: List[str], start_date: str = None, end_date: str = None ) -> pd.DataFrame: """Get Series data. [Source: FRED] Parameters @@ -197,8 +197,6 @@ def get_aggregated_series_data( Start date to get data from, format yyyy-mm-dd end_date : str End data to get from, format yyyy-mm-dd - limit: int - Number of raw data rows to show Returns ---------- @@ -229,9 +227,6 @@ def get_aggregated_series_data( axis=1, ) - if limit != -1: - data = data.tail(limit) - return data, detail diff --git a/openbb_terminal/economy/fred_view.py b/openbb_terminal/economy/fred_view.py index cf7c07e2d0e..8c7e48e3e2d 100644 --- a/openbb_terminal/economy/fred_view.py +++ b/openbb_terminal/economy/fred_view.py @@ -104,7 +104,7 @@ def display_fred_series( """ data, detail = fred_model.get_aggregated_series_data( - series_ids, start_date, end_date, limit + series_ids, start_date, end_date ) if data.empty: @@ -159,7 +159,7 @@ def display_fred_series( if raw: print_rich_table( - data, + data.tail(limit), headers=list(data.columns), show_index=True, index_name="Date",