code: def get_inference_data(self, eight_schools_params, predictions_data):
      predictions = predictions_data
      return from_beanmachine(
          sampler=predictions,
          coords={
              "school": np.arange(eight_schools_params["J"]),
              "school_pred": np.arange(eight_schools_params["J"]),
          },
      )
  
file_path: arviz/arviz/tests/external_tests/test_data_beanmachine.py
imports: import numpy
  import pytest
  from ...data.io_beanmachine import from_beanmachine
  from ..helpers import chains, draws, eight_schools_params, importorskip, load_cached_models
label_desc: Use image node_id 3 for calling the TestDataBeanMachine obj's underlying member method code with example usage: obj.get_inference_data(eight_schools_params, predictions_data) and returns: from_beanmachine