lgtm fixes

Signed-off-by: Evelina Bakhturina <[email protected]>
NVIDIA · Jun 3, 2020 · b752fec · b752fec
1 parent 307b942
commit b752fec
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py b/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py
@@ -176,9 +176,10 @@ def update_state(system_uttr, user_uttr, dial_history, belief_state):
 
 if args.mode == 'interactive':
     # for user_uttr in user_uttrs:
+    logging.info("============ Starting a new dialogue ============")
+    system_uttr, system_action, belief_state, dial_history = init_session()
     while True:
         logging.info("Type your text, use STOP to exit and RESTART to start a new dialogue.")
-        system_uttr, system_action, belief_state, dial_history = init_session()
         user_uttr = input()
 
         if user_uttr == "STOP":

diff --git a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py
@@ -27,7 +27,7 @@
 from nemo.backends.pytorch.nm import NonTrainableNM
 from nemo.collections.nlp.data.datasets.multiwoz_dataset.multiwoz_slot_trans import REF_SYS_DA
 from nemo.collections.nlp.utils.callback_utils import tensor2numpy
-from nemo.core import AxisKind, AxisType, LogitsType, NeuralType, StringType, VoidType
+from nemo.core import AxisKind, AxisType, LogitsType, NeuralType, StringType
 from nemo.utils import logging
 from nemo.utils.decorators import add_port_docs
 

diff --git a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py
@@ -21,7 +21,7 @@
 import torch
 
 from nemo.backends.pytorch.nm import NonTrainableNM
-from nemo.core import AxisKind, AxisType, ChannelType, LengthsType, NeuralType, StringType, VoidType
+from nemo.core import AxisKind, AxisType, ChannelType, LengthsType, NeuralType, StringType
 from nemo.utils import logging
 from nemo.utils.decorators import add_port_docs