Skip to content

Commit

Permalink
fix CI
Browse files Browse the repository at this point in the history
  • Loading branch information
peteryang1 committed Sep 20, 2024
1 parent 8dd3c8c commit c96f85a
Show file tree
Hide file tree
Showing 5 changed files with 6 additions and 8 deletions.
4 changes: 2 additions & 2 deletions rdagent/app/kaggle/loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from rdagent.components.workflow.conf import BasePropSetting
from rdagent.components.workflow.rd_loop import RDLoop
from rdagent.core.developer import Developer
from rdagent.core.exception import ModelEmptyError
from rdagent.core.exception import FactorEmptyError, ModelEmptyError
from rdagent.core.proposal import (
Hypothesis2Experiment,
HypothesisExperiment2Feedback,
Expand Down Expand Up @@ -71,7 +71,7 @@ def running(self, prev_out: dict[str, Any]):
logger.log_object(exp, tag="runner result")
return exp

skip_loop_error = (ModelEmptyError,)
skip_loop_error = (ModelEmptyError, FactorEmptyError)


def main(path=None, step_n=None, competition=None):
Expand Down
2 changes: 0 additions & 2 deletions rdagent/app/qlib_rd_loop/factor_from_report.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,6 @@
extract_first_page_screenshot_from_pdf,
load_and_process_pdfs_by_langchain,
)
from rdagent.components.workflow.rd_loop import RDLoop
from rdagent.core.exception import FactorEmptyError
from rdagent.core.prompts import Prompts
from rdagent.core.proposal import Hypothesis
from rdagent.log import rdagent_logger as logger
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import os

import pandas as pd
from sklearn.compose import ColumnTransformer
from sklearn.impute import SimpleImputer
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,7 @@
import xgboost as xgb



def select(X: pd.DataFrame) -> pd.DataFrame:
def select(X: pd.DataFrame) -> pd.DataFrame:
# Ignore feature selection logic
return X

Expand Down Expand Up @@ -38,4 +37,4 @@ def predict(model, X):
X = select(X)
dtest = xgb.DMatrix(X)
y_pred_prob = model.predict(dtest)
return y_pred_prob
return y_pred_prob
2 changes: 1 addition & 1 deletion rdagent/scenarios/kaggle/experiment/scenario.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import io
import json
from pathlib import Path
import pickle
from pathlib import Path

import pandas as pd
from jinja2 import Environment, StrictUndefined
Expand Down

0 comments on commit c96f85a

Please sign in to comment.