X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=ml%2Fquick_label.py;h=7e0a6bf64921533e00d719223d2657fb21ebbccf;hb=f2600f30801c849fc1d139386e3ddc3c9eb43e30;hp=1ed42966db65da792a81a802b8b5a606b55e4dcc;hpb=a838c154135b2420d9047a101caf24a2c9f593c2;p=python_utils.git

diff --git a/ml/quick_label.py b/ml/quick_label.py
index 1ed4296..7e0a6bf 100644
--- 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.