github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change base class's predict signature.

Browse Source
This commit is contained in:
h-alice 2024-12-31 14:06:33 +08:00
parent c31268463b
commit ffbba7fe83
No known key found for this signature in database
GPG Key ID: 5708F34144A70909
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/models/Demography.py
View File

@ -1,4 +1,4 @@
from typing import Union
from typing import Union, List
from abc import ABC, abstractmethod
import numpy as np
from deepface.commons import package_utils
@ -18,5 +18,5 @@ class Demography(ABC):
model_name: str
@abstractmethod
def predict(self, img: np.ndarray) -> Union[np.ndarray, np.float64]:
def predict(self, img: Union[np.ndarray, List[np.ndarray]]) -> Union[np.ndarray, np.float64]:
pass