microsoft
/
qlib
зеркало из https://github.com/microsoft/qlib.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

fix the warning (#1656)

This commit is contained in:
Andy li 2023-11-03 17:03:11 +08:00 коммит произвёл GitHub
Родитель fb80e318e2
Коммит 8bf2678676
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

4
scripts/dump_bin.py
Просмотреть файл

@ -135,7 +135,7 @@ class DumpDataBase:
def _get_source_data(self, file_path: Path) -> pd.DataFrame:
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)
return df
@ -195,7 +195,7 @@ class DumpDataBase:
def data_merge_calendar(self, df: pd.DataFrame, calendars_list: List[pd.Timestamp]) -> pd.DataFrame:
# calendars
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[
(calendars_df[self.date_field_name] >= df[self.date_field_name].min())
& (calendars_df[self.date_field_name] <= df[self.date_field_name].max())