From 8245faecff6f9a6bf7fa579ce03f2b905fc5da70 Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Tue, 30 Aug 2022 13:20:04 +0200 Subject: [PATCH] remove unwanted kwarg handover --- image_prediction/redai_adapter/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image_prediction/redai_adapter/model.py b/image_prediction/redai_adapter/model.py index a0ac3a8..05efc3a 100644 --- a/image_prediction/redai_adapter/model.py +++ b/image_prediction/redai_adapter/model.py @@ -12,7 +12,7 @@ class PredictionModelHandle: def predict(self, *args, **kwargs): tensor, valid_mask = self.__prep_images(*args, **kwargs) - predictions = self.__predict(tensor, **kwargs) + predictions = self.__predict(tensor) return [p if v else None for p, v in zip(predictions, valid_mask)] def __call__(self, *args, **kwargs):