Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
malfet committed Nov 6, 2024
1 parent d8ece6e commit 88590cd
Show file tree
Hide file tree
Showing 2 changed files with 524 additions and 280 deletions.
10 changes: 5 additions & 5 deletions aarch64_linux/aarch64_wheel_ci_build.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,19 @@

import os
import shutil
from subprocess import check_output, check_call
from pygit2 import Repository
from subprocess import check_call, check_output
from typing import List

from pygit2 import Repository


def list_dir(path: str) -> List[str]:
"""'
Helper for getting paths for Python
"""
return check_output(["ls", "-1", path]).decode().split("\n")


def build_ArmComputeLibrary() -> None:
"""
Using ArmComputeLibrary for aarch64 PyTorch
Expand Down Expand Up @@ -195,9 +197,7 @@ def parse_arguments():
)
if enable_cuda:
desired_cuda = os.getenv("DESIRED_CUDA")
build_vars += (
f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 "
)
build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 "
else:
build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1 "
elif branch.startswith(("v1.", "v2.")):
Expand Down
Loading

0 comments on commit 88590cd

Please sign in to comment.