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

diff --git a/ml/quick_label.py b/ml/quick_label.py
index 1ed4296..120ff5f 100644
--- a/ml/quick_label.py
+++ b/ml/quick_label.py
@@ -4,6 +4,7 @@ import glob
 import logging
 import os
 from typing import Callable, List, NamedTuple, Optional, Set
+import warnings
 
 import argparse_utils
 import config
@@ -90,9 +91,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.