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

Better type checking.

Browse Source
This commit is contained in:
Vincent STRAGIER 2023-06-23 17:37:19 +02:00
parent 1506a67725
commit 1a3e869737
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepface/DeepFace.py
View File

@ -296,17 +296,17 @@ def analyze(
if isinstance(actions, str): if isinstance(actions, str):
actions = (actions,) actions = (actions,)
actions = list(actions) # check if actions is not an iterable or empty.
if not hasattr(actions, "__getitem__") or not actions:
# Check if actions have been passed correctly
if not actions:
raise ValueError("`actions` must be a list of strings.") raise ValueError("`actions` must be a list of strings.")
actions = list(actions)
# For each action, check if it is valid # For each action, check if it is valid
for action in actions: for action in actions:
if action not in ("emotion", "age", "gender", "race"): if action not in ("emotion", "age", "gender", "race"):
raise ValueError( raise ValueError(
f"Invalid action passed ({action})). " f"Invalid action passed ({repr(action)})). "
"Valid actions are `emotion`, `age`, `gender`, `race`." "Valid actions are `emotion`, `age`, `gender`, `race`."
) )
# --------------------------------- # ---------------------------------