From e2d8dc2a404062532f7d82cbf00fc9d12421027a Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Fri, 18 Aug 2023 12:10:45 +0200
Subject: fuck poetry with torch

---
 pyproject.toml | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

(limited to 'pyproject.toml')

diff --git a/pyproject.toml b/pyproject.toml
index 928874a..9c09a99 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,13 +11,10 @@ packages = [{include = "swr2_asr"}]
 python = "^3.10"
 numpy = "^1.25.2"
 click = "^8.1.6"
-
-[tool.poetry.group.cpu]
-optional = true
-
-[tool.poetry.group.cpu.dependencies]
-torch = "=2.0.1"
-torchaudio = "=2.0.2"
+audioloader = {git = "https://github.com/KinWaiCheuk/AudioLoader.git"}
+tqdm = "^4.66.1"
+torch = {version = "^2.0.1+cpu", source = "pytorch-cpu"}
+torchaudio = {version = "^2.0.2+cpu", source = "pytorch-cpu"}
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.7.0"
@@ -26,18 +23,23 @@ mypy = "^1.5.1"
 pre-commit = "^3.3.3"
 
 [[tool.poetry.source]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu111/"
-priority = "supplemental"
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "explicit"
+
 
 [[tool.poetry.source]]
-name = "PyPI"
-priority = "primary"
+name = "pytorch-gpu"
+url = "https://download.pytorch.org/whl/cu118"
+priority = "explicit"
 
 [[tool.mypy.overrides]]
 module = "torchaudio.*"
 ignore_missing_imports = true
 
+[tool.poetry.scripts]
+train = "swr2_asr.train:main"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
-- 
cgit v1.2.3