code: def test_not_balance_model_data(model_data: RasaModelData):
      test_model_data = RasaModelData(
          label_key="entities",
          label_sub_key="tag_ids",
          data=model_data.data,
      )
  
      data = test_model_data.balanced_data(
          test_model_data.data, 2, False
      )
  
      assert np.all(
          data["entities"]["tag_ids"]
          == test_model_data.get("entities", "tag_ids")
      )
  
file_path: rasa/tests/utils/tensorflow/test_model_data.py
imports: import copy
  import pytest
  import numpy
  from rasa.utils.tensorflow.model_data import RasaModelData
label_desc: Use image node_id 11 for calling a global function with example usage: test_not_balance_model_data(model_data) without return types