move deprecation warning ro correct place

Granulate · Aug 1, 2024 · 3f46ea0 · 3f46ea0
1 parent 1cbc93e
commit 3f46ea0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/granulate_utils/linux/__init__.py b/granulate_utils/linux/__init__.py
@@ -25,11 +25,3 @@ def get_kernel_release() -> Tuple[int, int]:
 
 # TASK_COMM_LEN is 16, and it is always null-terminated, so 15.
 COMM_PATTERN = r".{0,15}"
-
-import warnings
-
-warnings.warn(
-    "granulate_utils.linux.cgroups is deprecated, use granulate_utils.linux.cgroups_v2 instead",
-    DeprecationWarning,
-    stacklevel=2,
-)
diff --git a/granulate_utils/linux/cgroups_v2/__init__.py b/granulate_utils/linux/cgroups_v2/__init__.py
@@ -14,6 +14,8 @@
 # limitations under the License.
 #
 
+import warnings
+
 from granulate_utils.linux.cgroups_v2.base_controller import BaseController  # noqa: F401
 from granulate_utils.linux.cgroups_v2.cgroup import get_process_cgroups  # noqa: F401
 from granulate_utils.linux.cgroups_v2.cpu_controller import CpuController, CpuControllerFactory  # noqa: F401
@@ -23,3 +25,9 @@
     SystemdLegacyController,
     SystemdUnifiedController,
 )
+
+warnings.warn(
+    "granulate_utils.linux.cgroups is deprecated, use granulate_utils.linux.cgroups_v2 instead",
+    DeprecationWarning,
+    stacklevel=2,
+)