Added torch dependency for running tests

ancestor-mithril · Sep 24, 2024 · cd41e2d · cd41e2d
1 parent 446d7a8
commit cd41e2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -27,7 +27,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install flake8 pytest
+        python -m pip install flake8 pytest torch
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Lint with flake8
       run: |

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,6 +31,7 @@ Issues = "https://github.com/ancestor-mithril/timed-decorator/issues"
 [project.optional-dependencies]
 dev = [
     "pytest",
-    "flake8"
+    "flake8",
+    "torch"
 ]