code: def call_spark_class(spark, class_name: str, args: str):
      spark_class = getattr(
          spark._jvm.com.facebook.spark.rl, class_name, None
      )
      assert spark_class is not None, f"Could not find {class_name}."
      spark_class.main(args)
  
file_path: ReAgent/reagent/data/spark_utils.py
imports: import logging
  import pprint
  from os.path import abspath, dirname, join
  from typing import Dict, Optional
  import reagent
  from pyspark.sql import SparkSession
  from pyspark.sql.functions import col
label_desc: Use image node_id 2 for calling a global function with example usage: call_spark_class(spark, class_name, args) without return types