code: def test_robust(spark_session):
      _run_job(
          spark=spark_session,
          name="robust",
          data_cols=["Weekly_Sales", "Temperature", "CPI"],
          hierarchical=True,
          agg_dict={
              "Weekly_Sales": "sum"
          },  # only use Weekly_Sales for the target
          predict_on_train=True,
          robust=True,
      )
  
file_path: merlion/tests/spark/test_forecast.py
imports: from os.path import abspath, dirname, join
  import logging
  from pyspark.sql.types import DateType, FloatType, StructField, StructType
  from merlion.spark.dataset import create_hier_dataset, read_dataset, write_dataset, TSID_COL_NAME
  from merlion.spark.pandas_udf import forecast, reconciliation
label_desc: Use image node_id 6 for calling a global function with example usage: test_robust(spark_session) without return types