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

added type hint

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-25 16:28:17 +01:00
parent 03f269c2d7
commit 2e36a9d46d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
image_prediction/service_estimator/service_estimator.py
View File

@ -1,5 +1,7 @@
from typing import Mapping, List
import numpy as np
from image_prediction.estimator.adapter.adapter import EstimatorAdapter
from image_prediction.utils import get_logger
@ -11,10 +13,10 @@ class ServiceEstimator:
self.__estimator_adapter = estimator_adapter
self.__classes = classes
def predict(self, batch) -> List[str]:
def predict(self, batch: np.array) -> List[str]:
if batch.shape[0] == 0:
logger.warning("ServiceEstimator received empty batch")
logger.warning("ServiceEstimator received empty batch.")
return []
return [self.__classes[numeric_label] for numeric_label in self.__estimator_adapter.predict(batch)]