fix: use a higher stacklevel on warnings

ai4os · Jun 7, 2024 · 8f3362d · 8f3362d
1 parent f89bc71
commit 8f3362d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/deepaas/api/v2/debug.py b/deepaas/api/v2/debug.py
@@ -71,7 +71,7 @@ def setup_debug():
             "restriction in the /debug/ endpoint. Disable it whenever "
             "you have finished debugging your application. \033[0m"
         )
-        warnings.warn(msg, RuntimeWarning)
+        warnings.warn(msg, RuntimeWarning, stacklevel=2)
 
         sys.stdout = MultiOut(DEBUG_STREAM, sys.stdout)
         sys.stderr = MultiOut(DEBUG_STREAM, sys.stderr)

diff --git a/deepaas/model/v2/__init__.py b/deepaas/model/v2/__init__.py
@@ -60,13 +60,14 @@ def register_models(app):
         warnings.warn(
             "Error loading models, using test model. This will be deprecated soon.",
             DeprecationWarning,
+            stacklevel=2,
         )
 
     if MODELS:
         if len(MODELS) > 1:
             # Loading several models will be deprecated in the future
             warn_msg = "Loading several models is deprecated."
-            warnings.warn(warn_msg, DeprecationWarning)
+            warnings.warn(warn_msg, DeprecationWarning, stacklevel=2)
             LOG.warning(warn_msg)
 
         MODELS_LOADED = True
@@ -79,7 +80,7 @@ def register_models(app):
             "API, please use the demo_app instead. "
             "Check https://github.com/deephdc/demo_app for more information.",
         )
-        warnings.warn(warn_msg, DeprecationWarning)
+        warnings.warn(warn_msg, DeprecationWarning, stacklevel=2)
         LOG.info("No models found in V2, loading test model")
         LOG.warning(warn_msg)
         MODELS["deepaas-test"] = wrapper.ModelWrapper(