diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 4281241b597..f6a12b94751 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -45,12 +45,12 @@ jobs: name: coverage-unit-data path: .coverage.* integrations: - name: Running framework integrations tests + name: framework-integration-tests runs-on: ubuntu-latest strategy: fail-fast: false matrix: - framework: [catboost, detectron, diffusers, easyocr, fastai, flax, keras, lightgbm, onnx, picklable_model, pytorch, pytorch_lightning, sklearn, tensorflow, torchscript, xgboost] + framework: [catboost, diffusers, easyocr, fastai, flax, keras, lightgbm, onnx, picklable_model, pytorch, pytorch_lightning, sklearn, tensorflow, torchscript, xgboost] timeout-minutes: 90 steps: - uses: actions/checkout@v3 diff --git a/tests/integration/frameworks/models/onnx.py b/tests/integration/frameworks/models/onnx.py index f7d4f4197a2..5e2b5048cdb 100644 --- a/tests/integration/frameworks/models/onnx.py +++ b/tests/integration/frameworks/models/onnx.py @@ -157,7 +157,7 @@ def make_rf_onnx_model() -> ( iris: sklearn.utils.Bunch = load_iris() X: ext.NpNDArray = iris.data y: ext.NpNDArray = iris.target - X_train, X_test, y_train, y_test = train_test_split(X, y) + X_train, X_test, y_train, _ = train_test_split(X, y) clr = RandomForestClassifier() clr.fit(X_train, y_train) @@ -200,7 +200,6 @@ def _check(out: tuple[ext.NpNDArray, list[dict[int, float]]]) -> bool: onnx_rf_model = FrameworkTestModel( name="onnx_rf_model", model=onnx_rf_raw_model, - model_method_caller=method_caller, model_signatures={"run": {"batchable": True}}, configurations=[ Config(