From bc758a67df597631f205805126fd835f420b2f85 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Mon, 15 Jul 2024 09:40:40 +0000 Subject: [PATCH] fix a small bug --- .../components/coder/factor_coder/CoSTEER/evaluators.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py b/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py index 9ead0c1d..6e19ddfa 100644 --- a/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py +++ b/rdagent/components/coder/factor_coder/CoSTEER/evaluators.py @@ -194,6 +194,14 @@ class FactorDatetimeDailyEvaluator(FactorEvaluator): if "datetime" not in gen_df.index.names: return "The source dataframe does not have a datetime index. Please check the implementation.", False + try: + pd.to_datetime(gen_df.index.get_level_values("datetime")) + except Exception: + return ( + "The source dataframe has a datetime index but it is not in the correct format (maybe a regular string or other objects). Please check the implementation.", + False, + ) + time_diff = gen_df.index.get_level_values("datetime").to_series().diff().dropna().unique() if pd.Timedelta(minutes=1) in time_diff: return (