diff --git a/deepaas/api/v2/debug.py b/deepaas/api/v2/debug.py index 012905ef..e723e3bb 100644 --- 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 index 920cc01b..bb82969d 100644 --- 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(