Style fix

deeppavlov · Dec 12, 2023 · 8d3ae9f · 8d3ae9f
1 parent 8d668bb
commit 8d3ae9f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/annotators/speech_function_classifier/models.py b/annotators/speech_function_classifier/models.py
@@ -6,14 +6,14 @@
 
 def check_sfs(predicted_sf, previous_sf, current_speaker, previous_speaker):
     if predicted_sf == "Command":
-        if ("Open" in previous_sf or previous_sf is '') and current_speaker == previous_speaker:
+        if ("Open" in previous_sf or previous_sf is "") and current_speaker == previous_speaker:
             return "Open.Command"
         elif current_speaker == previous_speaker:
             return "Sustain.Continue.Command"
         else:
             return "React.Respond.Command"
     elif predicted_sf == "Engage":
-        if previous_sf is '':
+        if previous_sf is "":
             return "Open.Attend"
         else:
             return "React.Respond.Support.Engage"

diff --git a/state_formatters/dp_formatters.py b/state_formatters/dp_formatters.py
@@ -1183,7 +1183,6 @@ def speech_function_predictor_formatter(dialog: Dict):
     res = None
     try:
         res = dialog["human_utterances"][-1]["annotations"].get("speech_function_classifier", [""])
-        # res = [item[0] for item in res]
         logger.info(f"formatter log sfp/sfc: {res}")
     except Exception as e:
         logger.error(f"Failed to format sfc output: {e}")