rasa-pro 3.11.5__py3-none-any.whl → 3.12.0__py3-none-any.whl

rasa/utils/tensorflow/layers.py

@@ -1,29 +1,35 @@
 import logging
-from typing import List, Optional, Text, Tuple, Callable, Union, Any
+from typing import Any, Callable, List, Optional, Text, Tuple, Union
+
 import tensorflow as tf
+import tensorflow.keras.backend as K
 
 # TODO: The following is not (yet) available via tf.keras
 from keras.src.utils.control_flow_util import smart_cond
-import tensorflow.keras.backend as K
 
 import rasa.utils.tensorflow.crf
+import rasa.utils.tensorflow.layers_utils as layers_utils
+from rasa.core.constants import DIALOGUE
+from rasa.shared.nlu.constants import (
+    ACTION_NAME,
+    ACTION_TEXT,
+    FEATURE_TYPE_SENTENCE,
+    FEATURE_TYPE_SEQUENCE,
+    INTENT,
+    TEXT,
+)
 from rasa.utils.tensorflow.constants import (
-    SOFTMAX,
-    MARGIN,
     COSINE,
-    INNER,
     CROSS_ENTROPY,
+    INNER,
     LABEL,
     LABEL_PAD_ID,
+    MARGIN,
+    SOFTMAX,
 )
-from rasa.core.constants import DIALOGUE
-from rasa.shared.nlu.constants import FEATURE_TYPE_SENTENCE, FEATURE_TYPE_SEQUENCE
-from rasa.shared.nlu.constants import TEXT, INTENT, ACTION_NAME, ACTION_TEXT
-
-from rasa.utils.tensorflow.metrics import F1Score
-from rasa.utils.tensorflow.exceptions import TFLayerConfigException
-import rasa.utils.tensorflow.layers_utils as layers_utils
 from rasa.utils.tensorflow.crf import crf_log_likelihood
+from rasa.utils.tensorflow.exceptions import TFLayerConfigException
+from rasa.utils.tensorflow.metrics import F1Score
 
 logger = logging.getLogger(__name__)
 

rasa/utils/tensorflow/layers_utils.py

@@ -1,6 +1,7 @@
+from typing import Optional, Union
+
 import tensorflow as tf
 from tensorflow import Tensor
-from typing import Union, Optional
 
 
 def random_indices(

rasa/utils/tensorflow/metrics.py

@@ -1,8 +1,8 @@
+from typing import Any, Dict, Optional
+
 import tensorflow as tf
 from tensorflow.keras import backend as K
 from tensorflow.types.experimental import TensorLike
-from typing import Any, Dict, Optional
-
 
 # original code taken from
 # https://github.com/tensorflow/addons/blob/f30df4322b5580b3e5946530a60f7126035dd73b/tensorflow_addons/metrics/f_scores.py

rasa/utils/tensorflow/model_data.py

@@ -1,20 +1,20 @@
 import logging
+from collections import OrderedDict, defaultdict
 from typing import (
-    Optional,
+    Any,
     DefaultDict,
     Dict,
+    ItemsView,
     Iterable,
-    Text,
     List,
+    NamedTuple,
+    Optional,
+    Text,
     Tuple,
-    Any,
     Union,
-    NamedTuple,
-    ItemsView,
-    overload,
     cast,
+    overload,
 )
-from collections import defaultdict, OrderedDict
 
 import numpy as np
 import scipy.sparse

rasa/utils/tensorflow/model_data_utils.py

@@ -1,25 +1,26 @@
-import typing
 import copy
+import typing
+from collections import OrderedDict, defaultdict
+from typing import Any, DefaultDict, Dict, List, Optional, Text, Tuple, Union, cast
+
 import numpy as np
 import scipy.sparse
-from collections import defaultdict, OrderedDict
-from typing import List, Optional, Text, Dict, Tuple, Union, Any, DefaultDict, cast
 
 from rasa.nlu.constants import TOKENS_NAMES
-from rasa.utils.tensorflow.model_data import Data, FeatureArray, ragged_array_to_ndarray
-from rasa.utils.tensorflow.constants import MASK, IDS
-from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.constants import (
-    TEXT,
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
+    TEXT,
 )
+from rasa.shared.nlu.training_data.message import Message
+from rasa.utils.tensorflow.constants import IDS, MASK
+from rasa.utils.tensorflow.model_data import Data, FeatureArray, ragged_array_to_ndarray
 
 if typing.TYPE_CHECKING:
-    from rasa.shared.nlu.training_data.features import Features
     from rasa.nlu.extractors.extractor import EntityTagSpec
+    from rasa.shared.nlu.training_data.features import Features
 
 TAG_ID_ORIGIN = "tag_id_origin"
 

rasa/utils/tensorflow/models.py

@@ -1,53 +1,52 @@
-import time
-import random
-import tensorflow as tf
-import numpy as np
 import logging
 import os
+import random
+import time
 from collections import defaultdict
-from typing import List, Text, Dict, Tuple, Union, Optional, Any, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Tuple, Union
 
-from keras.src.utils import tf_utils
+import numpy as np
+import tensorflow as tf
 from keras import Model
+from keras.src.utils import tf_utils
 
+import rasa.utils.train_utils
 from rasa.shared.constants import DIAGNOSTIC_DATA
+from rasa.shared.exceptions import RasaException
+from rasa.shared.nlu.constants import TEXT
+from rasa.utils.tensorflow import layers, rasa_layers
 from rasa.utils.tensorflow.constants import (
-    LABEL,
+    CONNECTION_DENSITY,
+    CONSTRAIN_SIMILARITIES,
+    EMBEDDING_DIMENSION,
     IDS,
     INTENT_CLASSIFICATION,
-    SENTENCE,
-    SEQUENCE_LENGTH,
-    RANDOM_SEED,
-    EMBEDDING_DIMENSION,
-    REGULARIZATION_CONSTANT,
-    SIMILARITY_TYPE,
-    CONNECTION_DENSITY,
-    NUM_NEG,
+    LABEL,
+    LEARNING_RATE,
     LOSS_TYPE,
-    MAX_POS_SIM,
     MAX_NEG_SIM,
-    USE_MAX_NEG_SIM,
-    NEGATIVE_MARGIN_SCALE,
-    SCALE_LOSS,
-    LEARNING_RATE,
-    CONSTRAIN_SIMILARITIES,
+    MAX_POS_SIM,
     MODEL_CONFIDENCE,
+    NEGATIVE_MARGIN_SCALE,
+    NUM_NEG,
+    RANDOM_SEED,
+    REGULARIZATION_CONSTANT,
     RUN_EAGERLY,
+    SCALE_LOSS,
+    SENTENCE,
+    SEQUENCE_LENGTH,
+    SIMILARITY_TYPE,
+    USE_MAX_NEG_SIM,
 )
-from rasa.utils.tensorflow.model_data import (
-    RasaModelData,
-    FeatureSignature,
-    FeatureArray,
-)
-import rasa.utils.train_utils
-from rasa.utils.tensorflow import layers
-from rasa.utils.tensorflow import rasa_layers
 from rasa.utils.tensorflow.data_generator import (
-    RasaDataGenerator,
     RasaBatchDataGenerator,
+    RasaDataGenerator,
+)
+from rasa.utils.tensorflow.model_data import (
+    FeatureArray,
+    FeatureSignature,
+    RasaModelData,
 )
-from rasa.shared.nlu.constants import TEXT
-from rasa.shared.exceptions import RasaException
 from rasa.utils.tensorflow.types import BatchData, MaybeNestedBatchData
 
 if TYPE_CHECKING:

rasa/utils/tensorflow/rasa_layers.py

@@ -1,34 +1,35 @@
-import tensorflow as tf
+from typing import Any, Callable, Dict, List, Optional, Text, Tuple, Union
+
 import numpy as np
-from typing import Text, List, Dict, Any, Union, Optional, Tuple, Callable
+import tensorflow as tf
 
+from rasa.nlu.constants import DEFAULT_TRANSFORMER_SIZE
 from rasa.shared.nlu.constants import TEXT
-from rasa.utils.tensorflow.model_data import FeatureSignature
+from rasa.utils.tensorflow import layers
 from rasa.utils.tensorflow.constants import (
-    REGULARIZATION_CONSTANT,
+    CONCAT_DIMENSION,
     CONNECTION_DENSITY,
+    DENSE_DIMENSION,
+    DENSE_INPUT_DROPOUT,
+    DROP_RATE,
+    DROP_RATE_ATTENTION,
+    HIDDEN_LAYERS_SIZES,
+    KEY_RELATIVE_ATTENTION,
+    MASKED_LM,
+    MAX_RELATIVE_POSITION,
+    NUM_HEADS,
     NUM_TRANSFORMER_LAYERS,
+    REGULARIZATION_CONSTANT,
+    SENTENCE,
+    SEQUENCE,
+    SPARSE_INPUT_DROPOUT,
     TRANSFORMER_SIZE,
-    NUM_HEADS,
     UNIDIRECTIONAL_ENCODER,
-    KEY_RELATIVE_ATTENTION,
     VALUE_RELATIVE_ATTENTION,
-    MAX_RELATIVE_POSITION,
-    MASKED_LM,
-    HIDDEN_LAYERS_SIZES,
-    DROP_RATE,
-    SPARSE_INPUT_DROPOUT,
-    DENSE_INPUT_DROPOUT,
-    DENSE_DIMENSION,
-    CONCAT_DIMENSION,
-    DROP_RATE_ATTENTION,
-    SEQUENCE,
-    SENTENCE,
 )
-from rasa.utils.tensorflow import layers
 from rasa.utils.tensorflow.exceptions import TFLayerConfigException
+from rasa.utils.tensorflow.model_data import FeatureSignature
 from rasa.utils.tensorflow.transformer import TransformerEncoder
-from rasa.nlu.constants import DEFAULT_TRANSFORMER_SIZE
 
 
 class RasaCustomLayer(tf.keras.layers.Layer):

rasa/utils/tensorflow/types.py

@@ -1,6 +1,7 @@
 from typing import Tuple, Union
-import tensorflow as tf
+
 import numpy as np
+import tensorflow as tf
 
 BatchData = Union[Tuple[tf.Tensor, ...], Tuple[np.ndarray, ...]]
 MaybeNestedBatchData = Union[Tuple[BatchData, ...], BatchData]

rasa/utils/train_utils.py

@@ -1,43 +1,45 @@
 from pathlib import Path
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Tuple, Union
+
 import numpy as np
-from typing import Optional, Text, Dict, Any, Union, List, Tuple, TYPE_CHECKING
 
+import rasa.nlu.utils.bilou_utils
 import rasa.shared.utils.common
 import rasa.shared.utils.io
-import rasa.nlu.utils.bilou_utils
-from rasa.shared.constants import NEXT_MAJOR_VERSION_FOR_DEPRECATIONS
-from rasa.nlu.constants import NUMBER_OF_SUB_TOKENS
 import rasa.utils.io as io_utils
+from rasa.nlu.constants import NUMBER_OF_SUB_TOKENS
+from rasa.shared.constants import NEXT_MAJOR_VERSION_FOR_DEPRECATIONS
+from rasa.shared.exceptions import InvalidConfigException
+from rasa.shared.nlu.constants import SPLIT_ENTITIES_BY_COMMA
+from rasa.utils.tensorflow.callback import RasaModelCheckpoint, RasaTrainingLogger
 from rasa.utils.tensorflow.constants import (
+    AUTO,
+    CHECKPOINT_MODEL,
+    CONSTRAIN_SIMILARITIES,
+    COSINE,
+    CROSS_ENTROPY,
+    EPOCHS,
+    EVAL_NUM_EPOCHS,
+    EVAL_NUM_EXAMPLES,
+    INNER,
     LOSS_TYPE,
+    MARGIN,
+    MODEL_CONFIDENCE,
     RANKING_LENGTH,
     RENORMALIZE_CONFIDENCES,
+    SEQUENCE,
     SIMILARITY_TYPE,
-    EVAL_NUM_EXAMPLES,
-    EVAL_NUM_EPOCHS,
-    EPOCHS,
     SOFTMAX,
-    MARGIN,
-    AUTO,
-    INNER,
-    COSINE,
-    SEQUENCE,
-    CROSS_ENTROPY,
-    CONSTRAIN_SIMILARITIES,
-    MODEL_CONFIDENCE,
     TOLERANCE,
-    CHECKPOINT_MODEL,
 )
-from rasa.utils.tensorflow.callback import RasaTrainingLogger, RasaModelCheckpoint
 from rasa.utils.tensorflow.data_generator import RasaBatchDataGenerator
 from rasa.utils.tensorflow.model_data import RasaModelData
-from rasa.shared.nlu.constants import SPLIT_ENTITIES_BY_COMMA
-from rasa.shared.exceptions import InvalidConfigException
 
 if TYPE_CHECKING:
+    from tensorflow.keras.callbacks import Callback
+
     from rasa.nlu.extractors.extractor import EntityTagSpec
     from rasa.nlu.tokenizers.tokenizer import Token
-    from tensorflow.keras.callbacks import Callback
 
 
 def rank_and_mask(
@@ -174,11 +176,11 @@ def update_evaluation_parameters(config: Dict[Text, Any]) -> Dict[Text, Any]:
 def load_tf_hub_model(model_url: Text) -> Any:
     """Load model from cache if possible, otherwise from TFHub."""
     import os
-    from tensorflow_hub.module_v2 import load as tfhub_load
 
     # needed to load the ConveRT model
     # noinspection PyUnresolvedReferences
     import tensorflow_text  # noqa: F401
+    from tensorflow_hub.module_v2 import load as tfhub_load
 
     # required to take care of cases when other files are already
     # stored in the default TFHUB_CACHE_DIR

rasa/utils/url_tools.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
+
 import enum
 from typing import Dict
-
 from urllib.parse import urlparse
 
 import structlog