diff --git a/sktalk/corpus/conversation.py b/sktalk/corpus/conversation.py index 2dd5a74..a6fbce4 100644 --- a/sktalk/corpus/conversation.py +++ b/sktalk/corpus/conversation.py @@ -79,4 +79,4 @@ def subconversation(self, index): def until_next(self): if len(self.utterances) != 2: raise ValueError("Conversation must have 2 utterances") - return self.utterances[0].until(self.utterances[1]) \ No newline at end of file + return self.utterances[0].until(self.utterances[1]) diff --git a/sktalk/corpus/utterance.py b/sktalk/corpus/utterance.py index 3590be7..9286789 100644 --- a/sktalk/corpus/utterance.py +++ b/sktalk/corpus/utterance.py @@ -47,6 +47,5 @@ def asdict(self): def until(self, next_utt): return next_utt.time[0] - self.time[1] - # TODO function: that prints summary of data, shows it to user # TODO function: create a pandas data frame with the utterances diff --git a/tests/conftest.py b/tests/conftest.py index d483521..1998a63 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -27,12 +27,12 @@ def convo_utts(): utterance1 = Utterance( utterance="Hello", participant="A", - time = [0, 1000] + time=[0, 1000] ) utterance2 = Utterance( utterance="Monde", participant="B", - time = [900, 1800] + time=[900, 1800] ) return [utterance1, utterance2] diff --git a/tests/corpus/test_conversation.py b/tests/corpus/test_conversation.py index 6ea0a82..394b5f3 100644 --- a/tests/corpus/test_conversation.py +++ b/tests/corpus/test_conversation.py @@ -48,4 +48,4 @@ def test_write_json(self, my_convo, tmp_path, user_path, expected_path): assert my_convo_read == my_convo.asdict() def test_until(self, my_convo): - assert my_convo.subconversation(0).until_next == -100 \ No newline at end of file + assert my_convo.subconversation(0).until_next == -100 diff --git a/tests/corpus/test_utterance.py b/tests/corpus/test_utterance.py index deb9cd6..ed26e66 100644 --- a/tests/corpus/test_utterance.py +++ b/tests/corpus/test_utterance.py @@ -31,4 +31,4 @@ def test_asdict(self): def test_until(self, convo_utts): utt1, utt2 = convo_utts - assert utt1.until(utt2) == -100 \ No newline at end of file + assert utt1.until(utt2) == -100