有时没有生成create_final_covariates.parquet。代码中需要兼容一下:
main-cn.py 将
covariate_df = pd.read_parquet(f"{INPUT_DIR}/{COVARIATE_TABLE}.parquet")
claims = read_indexer_covariates(covariate_df)
改为
final_covariates_path = f"{INPUT_DIR}/{COVARIATE_TABLE}.parquet"
final_covariates = (
pd.read_parquet(final_covariates_path)
if os.path.exists(final_covariates_path)
else None
)
claims = (
read_indexer_covariates(final_covariates)
if final_covariates is not None
else []
)