code: def find_wheel_package(dir):
      """Find the wheel package uploaded to this machine"""
      regular = re.compile("^nni-.*\.whl$")
      for file_name in os.listdir(dir):
          if regular.search(file_name):
              return file_name
      return None
  
file_path: nni/test/training_service/nnitest/remote_docker.py
imports: import os
  import argparse
  from subprocess import check_output, check_call
  import socket
  import random
  import re
label_desc: Use image node_id 3 for calling a global function with example usage: find_wheel_package(dir) and returns: None, file_name