remove unwanted kwarg handover

2022-08-30 13:20:04 +02:00 · 2022-08-30 13:20:04 +02:00 · 8245faecff
commit 8245faecff
parent 55a5dd11d6
1 changed files with 1 additions and 1 deletions
--- 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):