code: def spark_session():
      with _get_or_create_spark_session() as session:
          yield session
  
file_path: mlflow/tests/spark/autologging/datasource/test_spark_datasource_autologging_unit.py
imports: from unittest import mock
  import pytest
  import mlflow
  import mlflow.spark
  from mlflow._spark_autologging import PythonSubscriber, _get_current_listener
  from mlflow.exceptions import MlflowException
  from tests.spark.autologging.utils import _get_or_create_spark_session
label_desc: Use image node_id 1 for calling a global function with example usage: spark_session() without return types