code: def find_model_class(module):
      # find the first class that contents predict and train methods
      for _, klass in inspect.getmembers(module, inspect.isclass):
          funcs = [
              name
              for name, _ in inspect.getmembers(
                  klass, inspect.isfunction
              )
          ]
          if "predict" in funcs and "train" in funcs:
              return klass
  
file_path: mindsdb/mindsdb/integrations/handlers/byom_handler/proc_wrapper.py
imports: import io
  import sys
  import pickle
  import inspect
  from enum import Enum
  import pandas
label_desc: Use image node_id 8 for calling a global function with example usage: find_model_class(module) and returns: klass