code: def find_package_in_error_string(
      deps: List[str], line: str
  ) -> Optional[str]:
      # if the last word in the error string is in the list of deps, return it
      last_word = line.split(" ")[-1]
      if last_word in deps:
          return last_word
      # if the last word is not in the list of deps, check all words
      # TODO: this could report the wrong package if the error string
      # contains a reference to another package in the deps
      # before the package that failed to install
      for word in line.split(" "):
          if word.strip(",") in deps:
              return word
      # if we can't find the package, return None
      return None
  
file_path: wandb/wandb/sdk/launch/builder/templates/_wandb_bootstrap.py
imports: import json
  import os
  import re
  import subprocess
  import sys
  from typing import List, Optional, Set
label_desc: Use image node_id 5 for calling a global function with example usage: find_package_in_error_string(deps, line) and returns: None, last_word, word