Put pytest in testing group in poetry and update CI

Author: Peter Johnson

.github/workflows/staging-deploy.yml

@@ -59,7 +59,7 @@ jobs:
       - name: Install dependencies
         if: steps.dependencies-cache.outputs.cache-hit != 'true'
         run: |
-          poetry install  --no-interaction --no-root
+          poetry install  --with-test --no-interaction --no-root
 
       # TODO: add linting / black / flake8
       # - name: Lint with flake8

poetry.lock

@@ -19,9 +19,8 @@ priority = "explicit"
 python = ">=3.11,<3.14"
 
 # Uncomment below and comment out cpu if using macOS, the marker feature does not work (at least on my machine) - MM
-torch = { version = "^2.9.0" }
-#torch = { version = "^2.8.0+cpu", source = "pytorch-cpu" }
-
+#torch = { version = "^2.9.0", markers = "sys_platform != 'linux'" }
+torch = { version = "^2.8.0+cpu", markers = "sys_platform == 'linux'", source = "pytorch-cpu" }
 
 typing_extensions = "^4.15.0"
 lf_toolkit = { git = "https://github.com/lambda-feedback/toolkit-python.git", branch = "main", extras = [
@@ -38,12 +37,15 @@ sentencepiece = "^0.2.1"
 # https://github.com/lambda-feedback/toolkit-python/issues/2
 sympy = ">=1.14.0"
 
-[tool.poetry.group.dev.dependencies]
+[tool.poetry.group.test.dependencies]
 pytest = "^8.4.2"
+
+[tool.poetry.group.dev.dependencies]
 flake8 = "^7.3.0"
 nltk = "^3.9.2"
 matplotlib = "^3.10.7"
 pandas = "^2.3.3"
+graphviz = "^0.20.1"
 
 [build-system]
 requires = ["poetry-core"]