code: def tree_model() -> DecisionTreeClassificationModel:
      SparkSession.builder.master("local[2]").appName(
          "dtreeviz_sparkml"
      ).getOrCreate()
  
      spark_major_version = int(pyspark.__version__.split(".")[0])
      if spark_major_version >= 3:
          return DecisionTreeClassificationModel.load(
              "fixtures/spark_3_0_decision_tree_classifier.model"
          )
      elif spark_major_version >= 2:
          return DecisionTreeClassificationModel.load(
              "fixtures/spark_2_decision_tree_classifier.model"
          )
  
imports: import numpy
  import pyspark
  import pytest
  from pyspark.ml.classification import DecisionTreeClassificationModel
  from pyspark.sql import SparkSession
  from dtreeviz.models.spark_decision_tree import ShadowSparkTree
label_desc: Use image node_id 1 for calling a global function with example usage: tree_model() and returns: DecisionTreeClassificationModel, DecisionTreeClassificationModel