Skip to content

Commit

Permalink
Remove torch dependency.
Browse files Browse the repository at this point in the history
I think this is not needed anymore. Removing it also solves some
headaches with Poetry (see
python-poetry/poetry#6409) as well as allow
us to relax our Python interpreter version constraints.
  • Loading branch information
Sami Liedes committed Oct 5, 2023
1 parent c0ecdd2 commit 79d3697
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 178 deletions.
209 changes: 44 additions & 165 deletions poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

26 changes: 13 additions & 13 deletions pyproject.toml
Expand Up @@ -78,15 +78,22 @@ mark-parentheses = false
exempt-modules = ["typing", "typing_extensions"]

[tool.poetry]
name = "snowleo-llm"
name = "rocketrosti"
version = "0.1.0"
description = "Snow Leopard LLM"
authors = ["Sami Liedes <sami.liedes@rocket-science.ch>"]
readme = "README.md"
packages = [{include = "snowleo_llm"}]
packages = [{include = "rrosti"}]

[tool.poetry.dependencies]
python = ">=3.10,<3.11"

# Upper bound:
# We cannot have no upper bound because aiofile requires >=3.7, <4.
#
# We cannot have <4 because recent versions of scipy have an upperf bound of <3.13. Using <4 we get ancient versions
# of some packages.
python = ">=3.10,<3.13"

ipython = ">=8.13.1"
matplotlib = ">=3.7.1"
plotly = ">=5.14.1"
Expand All @@ -97,7 +104,8 @@ websockets = ">=11.0.2"
loguru = ">=0.7.1"

janus = ">=1.0.0"
pandas = ">=2.0.1"
pandas = ">=2.1.1"
scipy = ">=1.11.3"
openai = ">=0.27.6"
tabulate = ">=0.9.0"
orjson = ">=3.8.11"
Expand All @@ -106,14 +114,6 @@ joblib = ">=1.2.0"

# Hack to make sure we get the CPU only version. It's probably a good idea to update these to the
# latest version every now and then.
torch = [
{ markers = "sys_platform == 'linux'", url = "https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.0.1%2Bcpu.cxx11.abi-cp310-cp310-linux_x86_64.whl" },
{ markers = "sys_platform == 'win32'", url = "https://download.pytorch.org/whl/cpu/torch-2.0.1%2Bcpu-cp310-cp310-win_amd64.whl" },
]
torchvision = [
{ markers = "sys_platform == 'linux'", url = "https://download.pytorch.org/whl/cpu/torchvision-0.15.2%2Bcpu-cp310-cp310-linux_x86_64.whl"},
{ markers = "sys_platform == 'win32'", url = "https://download.pytorch.org/whl/cpu/torchvision-0.15.2%2Bcpu-cp310-cp310-win_amd64.whl"},
]
gevent = ">=22.10.2"
gspread = "^5.9.0"
aioconsole = "^0.6.1"
Expand Down Expand Up @@ -141,7 +141,7 @@ types-pyyaml = ">=6.0.12.9"
types-toml = ">=0.10.8.6"
types-beautifulsoup4 = ">=4.12.0.4"
types-tqdm = ">=4.65.0.1"
pandas-stubs = ">=2.0.1.230501"
pandas-stubs = ">=2.1.1.230928"
pytest = "^7.3.1"
pytest-mock = "^3.10.0"
pytest-asyncio = "^0.21.0"
Expand Down

0 comments on commit 79d3697

Please sign in to comment.