Skip to content

Commit

Permalink
Typing
Browse files Browse the repository at this point in the history
  • Loading branch information
WojciechBarczynski committed Nov 6, 2024
1 parent 25fac35 commit 2ed9ce3
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion child_lab_framework/task/emotions/emotions.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import asyncio
from typing import Dict, List
from deepface import DeepFace
from concurrent.futures import ThreadPoolExecutor
from itertools import repeat, starmap
Expand Down Expand Up @@ -85,7 +86,7 @@ async def stream(
case _:
results = None

def score_emotions(emotions):
def score_emotions(emotions: List[Dict[str, float]]) -> float:
# Most of the time, "angry" and "fear" are similar to "neutral" in the reality
scores = {
'angry': -0.05,
Expand Down

0 comments on commit 2ed9ce3

Please sign in to comment.