WIP

dagardner-nv · Aug 30, 2024 · 7ee095b · 7ee095b
1 parent fb9cc73
commit 7ee095b
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 29 deletions.
diff --git a/python/morpheus/morpheus/_lib/common/__init__.pyi b/python/morpheus/morpheus/_lib/common/__init__.pyi
@@ -5,11 +5,9 @@
            :toctree: _generate
         """
 from __future__ import annotations
-
-import os
-import typing
-
 import morpheus._lib.common
+import typing
+import os
 
 __all__ = [
     "FiberQueue",

diff --git a/python/morpheus/morpheus/_lib/cudf_helpers/__init__.pyi b/python/morpheus/morpheus/_lib/cudf_helpers/__init__.pyi
@@ -1,11 +1,8 @@
 from __future__ import annotations
-
+import morpheus._lib.cudf_helpers
 import typing
-
-import cudf
 from cudf.core.dtypes import StructDtype
-
-import morpheus._lib.cudf_helpers
+import cudf
 
 __all__ = [
     "StructDtype",

diff --git a/python/morpheus/morpheus/_lib/messages/__init__.pyi b/python/morpheus/morpheus/_lib/messages/__init__.pyi
@@ -6,14 +6,11 @@
 
         """
 from __future__ import annotations
-
+import morpheus._lib.messages
 import typing
-
 import cupy
-import mrc.core.node
-
 import morpheus._lib.common
-import morpheus._lib.messages
+import mrc.core.node
 
 __all__ = [
     "ControlMessage",

diff --git a/python/morpheus/morpheus/_lib/modules/__init__.pyi b/python/morpheus/morpheus/_lib/modules/__init__.pyi
@@ -6,10 +6,8 @@
 
         """
 from __future__ import annotations
-
-import typing
-
 import morpheus._lib.modules
+import typing
 
 __all__ = [
 

diff --git a/python/morpheus/morpheus/_lib/stages/__init__.pyi b/python/morpheus/morpheus/_lib/stages/__init__.pyi
@@ -6,16 +6,13 @@
 
         """
 from __future__ import annotations
-
-import os
+import morpheus._lib.stages
 import typing
-
+from morpheus._lib.common import FilterSource
+import morpheus._lib.common
 import mrc.core.coro
 import mrc.core.segment
-
-import morpheus._lib.common
-import morpheus._lib.stages
-from morpheus._lib.common import FilterSource
+import os
 
 __all__ = [
     "AddClassificationsControlMessageStage",

diff --git a/python/morpheus_llm/morpheus_llm/_lib/llm/__init__.pyi b/python/morpheus_llm/morpheus_llm/_lib/llm/__init__.pyi
@@ -6,15 +6,12 @@
 
         """
 from __future__ import annotations
-
+import morpheus_llm._lib.llm
 import typing
-
+import morpheus._lib.messages
 import mrc.core.coro
 import mrc.core.segment
 
-import morpheus._lib.llm
-import morpheus._lib.messages
-
 __all__ = [
     "InputMap",
     "LLMContext",