code: def get_branch_argument(version):
      """Get argument to branch.
  
      Before v0.1.8 it was --branch. From v0.1.8 and forward it is now --branch-to.
      """
      return (
          "--branch" if version in ["0.1.6", "0.1.7"] else "--branch-to"
      )
  
imports: import os
  import shutil
  import subprocess
  import pytest
  from pymongo import MongoClient
  import orion.core.cli
  import orion.core.io.experiment_builder
  from orion.client import create_experiment
  from orion.core.io.database import database_factory
  from orion.storage.base import storage_factory
label_desc: Use image node_id 2 for calling a global function with example usage: get_branch_argument(version) and returns: unknown