Skip to content

Commit

Permalink
fix typing nits
Browse files Browse the repository at this point in the history
  • Loading branch information
telamonian committed Sep 15, 2024
1 parent ccf2994 commit a9f954c
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions comfy_cli/standalone.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def clean(self):
for pycache in self.rpath.glob("**/__pycache__"):
shutil.rmtree(pycache)

def run_module(self, mod: str, *args: list[str]):
def run_module(self, mod: str, *args: str):
cmd: list[str] = [
str(self.executable),
"-m",
Expand All @@ -125,10 +125,10 @@ def run_module(self, mod: str, *args: list[str]):

subprocess.run(cmd, check=True)

def pip_install(self, *args: list[str]):
def pip_install(self, *args: str):
self.run_module("pip", "install", *args)

def uv_install(self, *args: list[str]):
def uv_install(self, *args: str):
self.run_module("uv", "pip", "install", *args)

def install_comfy_cli(self, dev: bool = False):
Expand All @@ -137,10 +137,10 @@ def install_comfy_cli(self, dev: bool = False):
else:
self.uv_install("comfy_cli")

def run_comfy_cli(self, *args: list[str]):
def run_comfy_cli(self, *args: str):
self.run_module("comfy_cli", *args)

def install_comfy(self, *args: list[str], gpu_arg: str = "--nvidia"):
def install_comfy(self, *args: str, gpu_arg: str = "--nvidia"):
self.run_comfy_cli("--here", "--skip-prompt", "install", "--fast-deps", gpu_arg, *args)

def dehydrate_comfy_deps(
Expand Down Expand Up @@ -171,7 +171,6 @@ def rehydrate_comfy_deps(self, packWheels: bool = False):
else:
self.dep_comp.install_deps()


def to_tarball(self, outPath: Optional[PathLike] = None, show_progress: bool = True):
# remove any __pycache__ before creating archive
self.clean()
Expand Down

0 comments on commit a9f954c

Please sign in to comment.