dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 00:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Removed hardcoded torch version and used the one from requirements.txt

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-09 16:08:44 +01:00
parent 1d2ade748c
commit 2af9876b76
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
setup.py
View file

@ -8,6 +8,13 @@ from setuptools import find_packages, setup
with open("requirements.txt") as f: with open("requirements.txt") as f:
requirements = f.read().splitlines() requirements = f.read().splitlines()
# Find the torch requirement and remove it from the list
torch = next((req for req in requirements if req.startswith("torch==")), None)
if torch:
requirements.remove(torch)
else:
torch = "torch==2.2.2" # Fallback if not found in requirements.txt
from aider import __version__ from aider import __version__
from aider.help_pats import exclude_website_pats from aider.help_pats import exclude_website_pats
@ -19,8 +26,6 @@ with open("README.md", "r", encoding="utf-8") as f:
# Debug: Print discovered packages # Debug: Print discovered packages
packages = find_packages(exclude=["benchmark"]) + ["aider.website"] packages = find_packages(exclude=["benchmark"]) + ["aider.website"]
print("Discovered packages:", packages) print("Discovered packages:", packages)
torch = "torch==2.2.2"
pytorch_url = None pytorch_url = None
with TemporaryDirectory(prefix="pytorch_download_") as temp_dir: with TemporaryDirectory(prefix="pytorch_download_") as temp_dir: