mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
Used TempNamedDir instead of tempfile.mkdtemp() to ensure the temporary directory is cleaned up after the subprocess completes.
This commit is contained in:
parent
472ad976b8
commit
f568c45f6b
1 changed files with 29 additions and 27 deletions
8
setup.py
8
setup.py
|
@ -1,7 +1,7 @@
|
|||
import re
|
||||
import subprocess
|
||||
import sys
|
||||
import tempfile
|
||||
from tempfile import TempNamedDir
|
||||
|
||||
from setuptools import find_packages, setup
|
||||
|
||||
|
@ -21,6 +21,9 @@ packages = find_packages(exclude=["benchmark"]) + ["aider.website"]
|
|||
print("Discovered packages:", packages)
|
||||
|
||||
torch = "torch==2.2.2"
|
||||
pytorch_url = None
|
||||
|
||||
with TempNamedDir(prefix="pytorch_download_") as temp_dir:
|
||||
cmd = [
|
||||
sys.executable,
|
||||
"-m",
|
||||
|
@ -29,12 +32,11 @@ cmd = [
|
|||
torch,
|
||||
"--no-deps",
|
||||
"--dest",
|
||||
tempfile.mkdtemp(prefix="pytorch_download_"),
|
||||
temp_dir.name,
|
||||
"--index-url",
|
||||
"https://download.pytorch.org/whl/cpu",
|
||||
]
|
||||
|
||||
pytorch_url = None
|
||||
try:
|
||||
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
|
||||
for line in process.stdout:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue