From 509eda073918af5c95b1d0a49f698611d4dd35e1 Mon Sep 17 00:00:00 2001 From: mandyche Date: Tue, 1 Aug 2023 14:58:36 -0400 Subject: [PATCH] add stateless case --- examples/speech_to_speech/english_alternate_agent.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/speech_to_speech/english_alternate_agent.py b/examples/speech_to_speech/english_alternate_agent.py index ce61d03c..c9761134 100644 --- a/examples/speech_to_speech/english_alternate_agent.py +++ b/examples/speech_to_speech/english_alternate_agent.py @@ -1,3 +1,5 @@ +from typing import Optional +from simuleval.agents.states import AgentStates from simuleval.utils import entrypoint from simuleval.data.segments import SpeechSegment from simuleval.agents import SpeechToSpeechAgent @@ -50,7 +52,10 @@ def __init__(self, args): def add_args(parser): parser.add_argument("--wait-seconds", default=1, type=int) - def policy(self): + def policy(self, states: Optional[AgentStates] = None): + if states is None: + states = states + length_in_seconds = round( len(self.states.source) / self.states.source_sample_rate )