Black

iai-group · Oct 16, 2023 · 16940b2 · 16940b2
1 parent c406c88
commit 16940b2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/moviebot/nlg/nlg.py b/moviebot/nlg/nlg.py
@@ -145,9 +145,7 @@ def __init__(self, args=None) -> None:
 
         self.slot_not_found = {
             Slots.GENRES.value: ["I could not find the genres __replace__."],
-            Slots.KEYWORDS.value: [
-                "I couldn't find the keywords __replace__."
-            ],
+            Slots.KEYWORDS.value: ["I couldn't find the keywords __replace__."],
             Slots.DIRECTORS.value: [
                 "I could not find the the director name __replace__."
             ],
@@ -520,9 +518,7 @@ def _clarify_CIN(  # noqa: C901
                     )
                 else:
                     response = (
-                        response
-                        + " named similar to "
-                        + CIN[Slots.TITLE.value]
+                        response + " named similar to " + CIN[Slots.TITLE.value]
                     )
         if (
             CIN[Slots.DIRECTORS.value]
@@ -580,9 +576,7 @@ def _user_options_continue(self, agent_dact: DialogueAct) -> ButtonOptions:
                 DialogueAct(UserIntents.RESTART, []): [
                     "I want to restart for a new movie."
                 ],
-                DialogueAct(UserIntents.BYE, []): [
-                    "I would like to quit now."
-                ],
+                DialogueAct(UserIntents.BYE, []): ["I would like to quit now."],
             }
             return options
 

diff --git a/tests/user_modeling/test_user_modeling.py b/tests/user_modeling/test_user_modeling.py
@@ -100,8 +100,7 @@ def test_get_utterances_with_slot_preferences(user_model: UserModel) -> None:
 
     assert user_model.get_utterances_with_slot_preferences("actors") == []
     assert (
-        user_model.get_utterances_with_slot_preferences("genre", "comedy")
-        == []
+        user_model.get_utterances_with_slot_preferences("genre", "comedy") == []
     )
     assert user_model.get_utterances_with_slot_preferences(
         "genre", "action"