code: def get_framework(test_class):
      """Infer the framework from the test class `test_class`."""
  
      if "ModelTesterMixin" in [
          x.__name__ for x in test_class.__bases__
      ]:
          return "pt"
      elif "TFModelTesterMixin" in [
          x.__name__ for x in test_class.__bases__
      ]:
          return "tf"
      elif "FlaxModelTesterMixin" in [
          x.__name__ for x in test_class.__bases__
      ]:
          return "flax"
      else:
          return None
  
file_path: transformers/utils/add_pipeline_model_mapping_to_test.py
imports: import argparse
  import glob
  import inspect
  import os
  import re
  import unittest
  from get_test_info import get_test_classes
  from tests.test_pipeline_mixin import pipeline_test_mapping
label_desc: Use image node_id 1 for calling a global function with example usage: get_framework(test_class) and returns: str, str, str, None