cleanup imports

rasa/core/policies/ensemble.py

@@ -7,8 +7,6 @@
 from datetime import datetime
 from typing import Text, Optional, Any, List, Dict, Tuple
 
-import numpy as np
-
 import rasa.core
 import rasa.utils.io
 from rasa.constants import MINIMUM_COMPATIBLE_VERSION, DOCS_BASE_URL
@@ -277,8 +275,10 @@ def load(cls, path: Text) -> "PolicyEnsemble":
         return ensemble
 
     @classmethod
-    def from_dict(cls, dictionary: Dict[Text, Any]) -> List[Policy]:
-        policies = dictionary.get("policies") or dictionary.get("policy")
+    def from_dict(cls, policy_configuration: Dict[Text, Any]) -> List[Policy]:
+        policies = policy_configuration.get("policies") or policy_configuration.get(
+            "policy"
+        )
         if policies is None:
             raise InvalidPolicyConfig(
                 "You didn't define any policies. "
@@ -373,6 +373,8 @@ def is_not_memo_policy(best_policy_name):
     def probabilities_using_best_policy(
         self, tracker: DialogueStateTracker, domain: Domain
     ) -> Tuple[Optional[List[float]], Optional[Text]]:
+        import numpy as np
+
         result = None
         max_confidence = -1
         best_policy_name = None

rasa/core/policies/keras_policy.py

@@ -19,7 +19,6 @@
 from rasa.core.policies.policy import Policy
 from rasa.core.trackers import DialogueStateTracker
 from rasa.utils.common import obtain_verbosity
-from rasa.utils.train_utils import load_tf_config
 from rasa.core.constants import DEFAULT_POLICY_PRIORITY
 
 # there are a number of issues with imports from tensorflow. hence the deactivation
@@ -78,6 +77,8 @@ def __init__(
         self.current_epoch = current_epoch
 
     def _load_params(self, **kwargs: Dict[Text, Any]) -> None:
+        from rasa.utils.train_utils import load_tf_config
+
         config = copy.deepcopy(self.defaults)
         config.update(kwargs)
 

rasa/core/policies/policy.py

@@ -1,6 +1,5 @@
 import copy
 import logging
-import tensorflow as tf
 from typing import Any, List, Optional, Text, Dict, Callable
 
 import rasa.utils.common

rasa/utils/train_utils.py

@@ -1,5 +1,7 @@
 from collections import namedtuple
 import logging
+import typing
+from typing import List, Optional, Text, Dict, Tuple, Union, Generator, Callable, Any
 import numpy as np
 from tqdm import tqdm
 from sklearn.model_selection import train_test_split
@@ -11,8 +13,7 @@
 )
 from tensor2tensor.layers.common_attention import large_compatible_negative
 from rasa.utils.common import is_logging_disabled
-import typing
-from typing import List, Optional, Text, Dict, Tuple, Union, Generator, Callable, Any
+
 
 if typing.TYPE_CHECKING:
     from tensor2tensor.utils.hparam import HParams