remove legacy environments commands and run black

mystic-ai · Jul 26, 2024 · 4b8ea10 · 4b8ea10
1 parent 85e6a03
commit 4b8ea10
Show file tree

Hide file tree

Showing 10 changed files with 6 additions and 40 deletions.
diff --git a/examples/audio-to-text/whisper-large/new_pipeline.py b/examples/audio-to-text/whisper-large/new_pipeline.py
@@ -24,8 +24,6 @@ class ModelKwargs(InputSchema):
 
 @entity
 class WhisperModel:
-    def __init__(self):
-        ...
 
     @pipe(on_startup=True, run_once=True)
     def load(self):

diff --git a/examples/image-to-image/t2i-adapter-sketch/new_pipeline.py b/examples/image-to-image/t2i-adapter-sketch/new_pipeline.py
@@ -108,8 +108,6 @@ class ModelKwargs(InputSchema):
 
 @entity
 class DiffusionWithAdapter:
-    def __init__(self) -> None:
-        ...
 
     def apply_style(
         self, style_name: str, positive: str, negative: str = ""

diff --git a/examples/text-to-audio/musicgen-large/new_pipeline.py b/examples/text-to-audio/musicgen-large/new_pipeline.py
@@ -4,8 +4,6 @@
 
 @entity
 class MusicgenModel:
-    def __init__(self):
-        ...
 
     @pipe(on_startup=True, run_once=True)
     def load(self):

diff --git a/pipeline/console/cluster.py b/pipeline/console/cluster.py
@@ -52,7 +52,7 @@ def use_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
 
 
 def remove_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
-    ...
+    pass
 
 
 def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:

diff --git a/pipeline/console/commands.py b/pipeline/console/commands.py
@@ -2,7 +2,6 @@
 
 from pipeline.console import cluster, container, logs
 from pipeline.console.targets import (
-    environments,
     files,
     pipelines,
     pointers,
@@ -23,7 +22,6 @@ def create_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
         dest="target",
     )
 
-    environments.create_parser(create_sub_parser)
     pointers.create_parser(create_sub_parser)
     resources.create_parser(create_sub_parser)
     files.create_parser(create_sub_parser)
@@ -42,7 +40,6 @@ def edit_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
         dest="target",
     )
 
-    environments.edit_parser(edit_sub_parser)
     pipelines.edit_parser(edit_sub_parser)
     pointers.edit_parser(edit_sub_parser)
     files.edit_parser(edit_sub_parser)
@@ -61,7 +58,6 @@ def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
         dest="target",
     )
 
-    environments.get_parser(get_sub_parser)
     pipelines.get_parser(get_sub_parser)
     pointers.get_parser(get_sub_parser)
     resources.get_parser(get_sub_parser)
@@ -82,7 +78,6 @@ def delete_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
         dest="target",
     )
 
-    environments.delete_parser(delete_sub_parser)
     pipelines.delete_parser(delete_sub_parser)
     pointers.delete_parser(delete_sub_parser)
     resources.delete_parser(delete_sub_parser)

diff --git a/pipeline/console/targets/environments.py b/pipeline/console/targets/environments.py
diff --git a/pipeline/console/targets/files.py b/pipeline/console/targets/files.py
@@ -41,7 +41,7 @@ def create_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
 
 
 def edit_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
-    ...
+    pass
 
 
 def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
@@ -72,8 +72,7 @@ def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
     )
 
 
-def delete_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
-    ...
+def delete_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None: ...
 
 
 def _create_file(args: Namespace) -> None:

diff --git a/pipeline/console/targets/resources.py b/pipeline/console/targets/resources.py
@@ -10,7 +10,7 @@
 
 
 def create_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
-    ...
+    pass
 
 
 def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
@@ -21,7 +21,7 @@ def get_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
 
 
 def delete_parser(command_parser: "_SubParsersAction[ArgumentParser]") -> None:
-    ...
+    pass
 
 
 # Functions

diff --git a/pipeline/objects/pipeline.py b/pipeline/objects/pipeline.py
@@ -6,11 +6,6 @@ class Pipeline:
     _current_pipeline: Graph
     _pipeline_context_active: bool = False
 
-    def __init__(
-        self,
-    ):
-        ...
-
     def __enter__(self):
         Pipeline._pipeline_context_active = True
 

diff --git a/tests/test_pipeline_function.py b/tests/test_pipeline_function.py
@@ -52,7 +52,7 @@ def test_function() -> Tuple[str, int]:
 
     @pipe
     def test_function_2(input_1: str):
-        ...
+        pass
 
     with pytest.raises(Exception):
         with Pipeline() as builder: