Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

remove unwanted kwarg handover

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-08-30 13:20:04 +02:00
parent 55a5dd11d6
commit 8245faecff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_prediction/redai_adapter/model.py
View File

@ -12,7 +12,7 @@ class PredictionModelHandle:
def predict(self, *args, **kwargs): def predict(self, *args, **kwargs):
tensor, valid_mask = self.__prep_images(*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)] return [p if v else None for p, v in zip(predictions, valid_mask)]
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):