liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00
Code Activity

fix the warning (#1656)

Browse Source
This commit is contained in:
Andy li
2023-11-03 17:03:11 +08:00
committed by GitHub
parent fb80e318e2
commit 8bf2678676
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/dump_bin.py
View File

@@ -135,7 +135,7 @@ class DumpDataBase:
def _get_source_data(self, file_path: Path) -> pd.DataFrame: def _get_source_data(self, file_path: Path) -> pd.DataFrame:
df = pd.read_csv(str(file_path.resolve()), low_memory=False) df = pd.read_csv(str(file_path.resolve()), low_memory=False)
df[self.date_field_name] = df[self.date_field_name].astype(str).astype(np.datetime64) df[self.date_field_name] = df[self.date_field_name].astype(str).astype("datetime64[ns]")
# df.drop_duplicates([self.date_field_name], inplace=True) # df.drop_duplicates([self.date_field_name], inplace=True)
return df return df
@@ -195,7 +195,7 @@ class DumpDataBase:
def data_merge_calendar(self, df: pd.DataFrame, calendars_list: List[pd.Timestamp]) -> pd.DataFrame: def data_merge_calendar(self, df: pd.DataFrame, calendars_list: List[pd.Timestamp]) -> pd.DataFrame:
# calendars # calendars
calendars_df = pd.DataFrame(data=calendars_list, columns=[self.date_field_name]) calendars_df = pd.DataFrame(data=calendars_list, columns=[self.date_field_name])
calendars_df[self.date_field_name] = calendars_df[self.date_field_name].astype(np.datetime64) calendars_df[self.date_field_name] = calendars_df[self.date_field_name].astype("datetime64[ns]")
cal_df = calendars_df[ cal_df = calendars_df[
(calendars_df[self.date_field_name] >= df[self.date_field_name].min()) (calendars_df[self.date_field_name] >= df[self.date_field_name].min())
& (calendars_df[self.date_field_name] <= df[self.date_field_name].max()) & (calendars_df[self.date_field_name] <= df[self.date_field_name].max())