fix: onnx tests

Signed-off-by: Aaron <[email protected]>
bentoml · Jul 17, 2023 · e5f49dd · e5f49dd
1 parent 5770008
commit e5f49dd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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(