def get_calibration_files()

in machine_learning/ml_infrastructure/inference-server-performance/server/scripts/tensorrt-optimization.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

def get_calibration_files(data_dir, filename_pattern):
  """We verify that the correct Imagenet data folder has been mounted
  and validation data files of the form validation-00xxx-of-00128 are
  available.
  """
  if data_dir is None:
    return []
  files = tf.gfile.Glob(os.path.join(data_dir, filename_pattern))
  if files == []:
    raise ValueError('Can not find any files in {} with '
                     'pattern "{}"'.format(data_dir, filename_pattern))
  return files