From d245242f2f859da841c6de70a2091f0cc78e9421 Mon Sep 17 00:00:00 2001 From: zhupr Date: Thu, 18 Mar 2021 11:21:25 +0800 Subject: [PATCH] Fix dump_bin.py && check_dump_bin.py --- scripts/check_dump_bin.py | 3 ++- scripts/dump_bin.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/check_dump_bin.py b/scripts/check_dump_bin.py index 7c2e837af..ef8023219 100644 --- a/scripts/check_dump_bin.py +++ b/scripts/check_dump_bin.py @@ -66,7 +66,7 @@ class CheckBin: self.csv_files = sorted(csv_path.glob(f"*{file_suffix}") if csv_path.is_dir() else [csv_path]) if check_fields is None: - check_fields = list(map(lambda x: x.split(".")[0], bin_path_list[0].glob(f"*.bin"))) + check_fields = list(map(lambda x: x.name.split(".")[0], bin_path_list[0].glob(f"*.bin"))) else: check_fields = check_fields.split(",") if isinstance(check_fields, str) else check_fields self.check_fields = list(map(lambda x: x.strip(), check_fields)) @@ -91,6 +91,7 @@ class CheckBin: origin_df[self.symbol_field_name] = symbol origin_df.set_index([self.symbol_field_name, self.date_field_name], inplace=True) origin_df.index.names = qlib_df.index.names + origin_df = origin_df.reindex(qlib_df.index) try: compare = datacompy.Compare( origin_df, diff --git a/scripts/dump_bin.py b/scripts/dump_bin.py index 4811fd486..38dd1f3af 100644 --- a/scripts/dump_bin.py +++ b/scripts/dump_bin.py @@ -219,7 +219,7 @@ class DumpDataBase: # used when creating a bin file date_index = self.get_datetime_index(_df, calendar_list) for field in self.get_dump_fields(_df.columns): - bin_path = features_dir.joinpath(f"{field}.{self.freq}{self.DUMP_FILE_SUFFIX}") + bin_path = features_dir.joinpath(f"{field.lower()}.{self.freq}{self.DUMP_FILE_SUFFIX}") if field not in _df.columns: continue if bin_path.exists() and self._mode == self.UPDATE_MODE: