code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument("--cluster-id", required=True)
      return parser.parse_args()
  
file_path: mlflow/examples/databricks/dbconnect.py
imports: import argparse
  from databricks.connect import DatabricksSession
  from databricks.sdk import WorkspaceClient
  from pyspark.sql.types import DoubleType
  from sklearn import datasets
  from sklearn.neighbors import KNeighborsClassifier
  import mlflow
  from mlflow.models import infer_signature
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: parser