code: def get_complete_for_one_or_zero(input):
      """Get shell complete for nargs='?'. Now only support zsh."""
      output = {}
      for sh, complete in input.items():
          if sh == "zsh":
              output[sh] = (
                  "_complete_for_one_or_zero '" + complete + "'"
              )
      return output
  
file_path: gpustat/gpustat/cli.py
imports: import os
  import sys
  import time
  from contextlib import suppress
  from blessed import Terminal
  from gpustat import __version__
  from gpustat.core import GPUStatCollection
label_desc: Use image node_id 2 for calling a global function with example usage: get_complete_for_one_or_zero(input) and returns: output