code: def add_python_extension_if_not_there(arg_string: str) -> str:
      """
      >>> # Setup
      >>> arg_string_with_py = __file__
      >>> arg_string_without_py = __file__.rsplit('.py',1)[0]
  
      >>> # Test with and without .py suffix
      >>> assert add_python_extension_if_not_there(arg_string_with_py) == __file__
      >>> assert add_python_extension_if_not_there(arg_string_without_py) == __file__
  
      """
  
      if not arg_string.endswith(".py"):
          arg_string = arg_string + ".py"
      return arg_string
  
file_path: neptune-client/src/neptune/vendor/lib_programname.py
imports: import pathlib
  import sys
  import __main__
label_desc: Use image node_id 6 for calling a global function with example usage: add_python_extension_if_not_there(arg_string) and returns: arg_string