code: def make_override_action(override_args):
      class StoreOverrideAction(argparse.Action):
          def __init__(
              self,
              option_strings,
              dest,
              default=None,
              type=None,
              choices=None,
              required=False,
              help=None,
          ):
              super(StoreOverrideAction, self).__init__(
                  option_strings=option_strings,
                  dest=dest,
                  nargs=1,
                  default=default,
                  type=type,
                  choices=choices,
                  required=required,
                  help=help,
              )
  
          def __call__(self, parser, args, values, option_string=None):
              override_args.add(self.dest)
              setattr(args, self.dest, values[0])
  
      return StoreOverrideAction
  
file_path: bluefog/bluefog/run/run.py
imports: import argparse
  import os
  import re
  import shlex
  import socket
  import subprocess
  import sys
  import traceback
  import psutil
  import bluefog
  from bluefog.run import env_util, network_util, horovod_driver
label_desc: Use image node_id 1 for calling a global function with example usage: make_override_action(override_args) and returns: StoreOverrideAction