Moving warning filter to the whole file.

radical-cybertools · Dec 11, 2020 · 79d5d97 · 79d5d97
1 parent 573cb41
commit 79d5d97
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/radical/entk/task.py b/src/radical/entk/task.py
@@ -7,6 +7,9 @@
 from . import exceptions as ree
 from . import states     as res
 
+import warnings
+warnings.simplefilter(action="once", category=DeprecationWarning, lineno=707)
+warnings.simplefilter(action="once", category=DeprecationWarning, lineno=764)
 
 # ------------------------------------------------------------------------------
 #
@@ -701,10 +704,10 @@ def cpu_reqs(self, value):
                              'cpu_process_type', 'cpu_thread_type'])
 
         if set(value.keys()).issubset(depr_expected_keys):
-            import warnings
-            warnings.simplefilter("once")
-            warnings.warn("CPU requirements keys are renamed using 'cpu_'" +
-                           "as a prefix for all keys.",DeprecationWarning)
+            warnings.warn("CPU requirements keys are renamed. Please use " +
+                          "cpu_processes for processes, cpu_process_type for " +
+                          "process_type, cpu_threads for threads_per_process " +
+                          "and cpu_thread_type for thread_type",DeprecationWarning)
 
             value['cpu_processes'] = value.pop('processes')
             value['cpu_process_type'] = value.pop('process_type')
@@ -758,10 +761,10 @@ def gpu_reqs(self, value):
                              'gpu_process_type', 'gpu_thread_type'])
 
         if set(value.keys()).issubset(depr_expected_keys):
-            import warnings
-            warnings.simplefilter("once")
-            warnings.warn("GPU requirements keys are renamed using 'gpu_'" +
-                           "as a prefix for all keys.",DeprecationWarning)
+            warnings.warn("GPU requirements keys are renamed. Please use " +
+                          "gpu_processes for processes, gpu_process_type for " +
+                          "process_type, gpu_threads for threads_per_process " +
+                          "and gpu_thread_type for thread_type",DeprecationWarning)
 
             value['gpu_processes'] = value.pop('processes')
             value['gpu_process_type'] = value.pop('process_type')