fix: lint

Signed-off-by: Aaron <[email protected]>
bentoml · Apr 1, 2023 · 3393fe7 · 3393fe7
1 parent 08d09ce
commit 3393fe7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integration/frameworks/models/transformers.py b/tests/integration/frameworks/models/transformers.py
@@ -429,7 +429,9 @@ def caller(
         model_inputs = kwargs
         if args:
             model_inputs.update(
-                t.cast("dict[str, t.Any]", vit_processor(arg, return_tensors=framework))
+                t.cast(
+                    "dict[str, t.Any]", vit_processor(args, return_tensors=framework)
+                )
             )
         last_hidden_state = getattr(framework_test_model.model, method)(
             **model_inputs