code: def main():
      parser = argparse.ArgumentParser()
      parser.add_argument("masked", choices=masked_packages)
      args = parser.parse_args()
      if args.masked == "torch":
          # https://stackoverflow.com/questions/1350466/preventing-python-code-from-importing-certain-modules
          sys.modules["torch"] = None
          import_related("tensorflow")
      elif args.masked == "torch_none":
          sys.modules["torch"] = None
          import_related("none")
      elif args.masked == "tensorflow":
          sys.modules["tensorflow"] = None
          import_related("pytorch")
      elif args.masked == "tianshou":
          sys.modules["tianshou"] = None
          import_rl_strategy_without_tianshou()
      else:
          raise ValueError(f"Unknown masked package: {args.masked}")
  
file_path: nni/test/ut/nas/test_optional_dependency.py
imports: import argparse
  import subprocess
  import sys
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: main() without return types