From ffbba7fe83c81c880ace6724d91b008428cb6b1f Mon Sep 17 00:00:00 2001
From: h-alice <admin@halice.art>
Date: Tue, 31 Dec 2024 14:06:33 +0800
Subject: [PATCH] Change base class's predict signature.

---
 deepface/models/Demography.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py
index ad93920..1dcef41 100644
--- a/deepface/models/Demography.py
+++ b/deepface/models/Demography.py
@@ -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