More cleanup.

[python_utils.git] / ml / quick_label.py

diff --git a/ml/quick_label.py b/ml/quick_label.py
index 1ed42966db65da792a81a802b8b5a606b55e4dcc..7e0a6bf64921533e00d719223d2657fb21ebbccf 100644 (file)
--- a/ml/quick_label.py
+++ b/ml/quick_label.py
@@ -3,6 +3,7 @@
 import glob
 import logging
 import os
+import warnings
 from typing import Callable, List, NamedTuple, Optional, Set
 
 import argparse_utils
@@ -79,9 +80,7 @@ def label(in_spec: InputSpec) -> None:
     elif in_spec.image_file_prepopulated_list is not None:
         images += in_spec.image_file_prepopulated_list
     else:
-        raise ValueError(
-            'One of image_file_glob or image_file_prepopulated_list is required'
-        )
+        raise ValueError('One of image_file_glob or image_file_prepopulated_list is required')
 
     skip_list = read_skip_list()
     for image in images:
@@ -90,9 +89,9 @@ def label(in_spec: InputSpec) -> None:
             continue
         features = in_spec.image_file_to_features_file(image)
         if features is None or not os.path.exists(features):
-            logger.warning(
-                f'File {image} yielded file {features} which does not exist, SKIPPING.'
-            )
+            msg = f'File {image} yielded file {features} which does not exist, SKIPPING.'
+            logger.warning(msg)
+            warnings.warn(msg)
             continue
 
         # Render features and image.