From 76c5328dddd545baf4956da09338ba23b51ae431 Mon Sep 17 00:00:00 2001 From: Linlang <30293408+SunsetWolf@users.noreply.github.com> Date: Sun, 11 Dec 2022 12:05:54 +0800 Subject: [PATCH] optimize_yahoo_collector (#1388) --- scripts/data_collector/yahoo/collector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/data_collector/yahoo/collector.py b/scripts/data_collector/yahoo/collector.py index 95a286b45f..143fa12ac2 100644 --- a/scripts/data_collector/yahoo/collector.py +++ b/scripts/data_collector/yahoo/collector.py @@ -222,8 +222,8 @@ def download_index_data(self): # TODO: from MSN _format = "%Y%m%d" _begin = self.start_datetime.strftime(_format) - _end = (self.end_datetime + pd.Timedelta(days=-1)).strftime(_format) - for _index_name, _index_code in {"csi300": "000300", "csi100": "000903"}.items(): + _end = self.end_datetime.strftime(_format) + for _index_name, _index_code in {"csi300": "000300", "csi100": "000903", "csi500": "000905"}.items(): logger.info(f"get bench data: {_index_name}({_index_code})......") try: df = pd.DataFrame(