code: def main():
      # Uncomment the following lines if you're running this script outside of Databricks
      # using a personal access token:
      # mlflow.set_tracking_uri("databricks")
      # mlflow.set_experiment("/Users/<username>/benchmark")
  
      FILE_SIZE = 1 * GiB
      NUM_FILES = 2
      NUM_ATTEMPTS = 3
  
      show_system_info()
      stats = []
      for i in range(NUM_ATTEMPTS):
          print(f"Attempt {i + 1} / {NUM_ATTEMPTS}")
          stats.append(upload_and_download(FILE_SIZE, NUM_FILES))
  
      df = pd.DataFrame(stats, columns=["upload [s]", "download [s]"])
      # show mean, min, max in markdown table
      print(df.aggregate(["count", "mean", "min", "max"]).to_markdown())
  
file_path: mlflow/examples/databricks/multipart.py
imports: import hashlib
  import json
  import os
  import pathlib
  import tempfile
  from concurrent.futures import ThreadPoolExecutor, as_completed
  import pandas
  import psutil
  from tqdm.auto import tqdm
  import mlflow
  from mlflow.environment_variables import MLFLOW_ENABLE_MULTIPART_DOWNLOAD, MLFLOW_ENABLE_MULTIPART_UPLOAD
  from mlflow.utils.time import Timer
label_desc: Use image node_id 7 for calling a global function with example usage: main() without return types