Embedding intent classifier breaks in rasa v1.7.2

The embedding intent classifier works fine in v1.7.0, but breaks in rasa v1.7.2

Here are the logs.

2020-02-15 09:37:57 INFO     rasa.nlu.model  - Finished training component.
2020-02-15 09:37:57 INFO     rasa.nlu.model  - Starting to train component EmbeddingIntentClassifier
2020-02-15 09:37:57 DEBUG    rasa.nlu.classifiers.embedding_intent_classifier  - Started training embedding classifier.
Epochs:   0%|                                                                                                   | 0/50 [00:00<?, ?it/s]Traceback (most recent call last):
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1365, in _do_call
    return fn(*args)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1350, in _run_fn
    target_list, run_metadata)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: ValueError: Number of examples differs for keys 'dict_keys(['text_features', 'label_features', 'label_ids', 'text_mask', 'label_mask'])'. Number of examples should be the same for all data in session data.
Traceback (most recent call last):

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/ops/script_ops.py", line 235, in __call__
    ret = func(*args)

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/data/ops/dataset_ops.py", line 594, in generator_py_func
    values = next(generator_state.get_iterator(iterator_id))

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/utils/train_utils.py", line 307, in gen_batch
    session_data = shuffle_session_data(session_data)

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/utils/train_utils.py", line 176, in shuffle_session_data
    data_points = get_number_of_examples(session_data)

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/utils/train_utils.py", line 290, in get_number_of_examples
    f"Number of examples differs for keys '{session_data.keys()}'. Number of "

ValueError: Number of examples differs for keys 'dict_keys(['text_features', 'label_features', 'label_ids', 'text_mask', 'label_mask'])'. Number of examples should be the same for all data in session data.


         [[{{node PyFunc}}]]
         [[IteratorGetNext]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/karthk/anaconda3/envs/rasa/bin/rasa", line 8, in <module>
    sys.exit(main())
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/__main__.py", line 76, in main
    cmdline_arguments.func(cmdline_arguments)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/cli/train.py", line 76, in train
    additional_arguments=extract_additional_arguments(args),
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/train.py", line 50, in train
    additional_arguments=additional_arguments,
  File "uvloop/loop.pyx", line 1456, in uvloop.loop.Loop.run_until_complete
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/train.py", line 101, in train_async
    additional_arguments,
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/train.py", line 188, in _train_async_internal
    additional_arguments=additional_arguments,
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/train.py", line 245, in _do_training
    persist_nlu_training_data=persist_nlu_training_data,
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/train.py", line 474, in _train_nlu_with_validated_data
    persist_nlu_training_data=persist_nlu_training_data,
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/nlu/train.py", line 86, in train
    interpreter = trainer.train(training_data, **kwargs)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/nlu/model.py", line 191, in train
    updates = component.train(working_data, self.config, **context)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/nlu/classifiers/embedding_intent_classifier.py", line 788, in train
    self.evaluate_every_num_epochs,
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/rasa/utils/train_utils.py", line 1193, in train_tf_dataset
    [train_op, loss, acc], feed_dict={is_training: True}
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 956, in run
    run_metadata_ptr)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1180, in _run
    feed_dict_tensor, options, run_metadata)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1359, in _do_run
    run_metadata)
  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1384, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: ValueError: Number of examples differs for keys 'dict_keys(['text_features', 'label_features', 'label_ids', 'text_mask', 'label_mask'])'. Number of examples should be the same for all data in session data.
Traceback (most recent call last):

  File "/home/karthk/anaconda3/envs/rasa/lib/python3.7/site-packages/tensorflow_core/python/ops/script_ops.py", line 235, in __call__
    ret = func(*args)

Hi @KarthiAru! Can you create an issue on GitHub in the Rasa repo for this?