Fix warning stacklevel

mitre · Apr 26, 2023 · 3c5b8e5 · 3c5b8e5
1 parent 955bb2a
commit 3c5b8e5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/service/data_svc.py b/app/service/data_svc.py
@@ -242,15 +242,15 @@ async def load_requirements_from_list(self, requirements: list):
         return [RequirementSchema().load(entry) for entry in requirements]
 
     async def load_adversary_file(self, filename, access):
-        warnings.warn(DEPRECATION_WARNING_LOAD)
+        warnings.warn(DEPRECATION_WARNING_LOAD, DeprecationWarning, stacklevel=2)
         await self.load_yaml_file(Adversary, filename, access)
 
     async def load_source_file(self, filename, access):
-        warnings.warn(DEPRECATION_WARNING_LOAD)
+        warnings.warn(DEPRECATION_WARNING_LOAD, DeprecationWarning, stacklevel=2)
         await self.load_yaml_file(Source, filename, access)
 
     async def load_objective_file(self, filename, access):
-        warnings.warn(DEPRECATION_WARNING_LOAD)
+        warnings.warn(DEPRECATION_WARNING_LOAD, DeprecationWarning, stacklevel=2)
         await self.load_yaml_file(Objective, filename, access)
 
     async def load_yaml_file(self, object_class, filename, access):