add a dependency group for finetuning

this will be ignored in the docker image
OpenFn · May 15, 2024 · d55723f · d55723f
1 parent f0e6af1
commit d55723f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 40 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,16 +9,20 @@ package-mode = false
 
 [tool.poetry.dependencies]
 python = "^3.11"
-requests = "^2.31.0"
-fastapi = "^0.110.2"
 spacy = "^3.7.4"
 en-core-web-sm = { path = "models/en_core_web_sm-3.7.1-py3-none-any.whl" }
 openai = "^1.23.3"
 transformers = "^4.40.1"
-torch = "^2.3.0"
 black = "^24.4.2"
 python-dotenv = "^1.0.1"
 
+# These dependencies should only be needed for fine tuning
+# use poetry install --with ft
+[tool.poetry.group.ft]
+optional = true
+
+[tool.poetry.group.ft.dependencies]
+torch = "^2.3.0"
 
 [build-system]
 requires = ["poetry-core"]