code: def test_sequence_tagger_softmax(self):
      model = SequenceTagger(
          num_pos_labels=5, num_chunk_labels=10, word_vocab_size=150
      )
      input_data = np.random.randint(150, size=(10, 50))
      output = model.predict(input_data, distributed=True)
      assert isinstance(output, list) and len(output) == 2
      assert output[0].shape == (10, 50, 5)
      assert output[1].shape == (10, 50, 10)
      self.assert_tfpark_model_save_load(model, input_data)
  
file_path: analytics-zoo/pyzoo/test/zoo/tfpark/test_text_models.py
imports: import pytest
  import numpy
  from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.tfpark.text.keras import *
label_desc: Use image node_id 5 for calling the TestTextModels obj's underlying member method code with example usage: obj.test_sequence_tagger_softmax() without return types