[CI] Address last comments

.github/workflows/unit_test.yaml

@@ -25,10 +25,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install dependencies
         run: |
-          python -m pip install -e ".[dev]"
-          # Will have to pin until monarch wheel is reasonable
-          pip install torch==2.9.0.dev20250815+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
-          python -m pip install --no-build-isolation --verbose assets/wheels/monarch_no_torch-0.1.0.dev20250815-py3-none-any.whl
+          python -m pip install --no-build-isolation -e ".[dev,oss]"
       - name: Run unit tests with coverage
         # TODO add all tests
         run: pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv

pyproject.toml

@@ -43,18 +43,14 @@ dev = [
     "pytest-asyncio",
 ]
 oss = [
-    "torch",
-    "torchmonarch-nightly==2025.8.1",
-    "torchstore",
+    "torch==2.9.0.dev20250815+cpu",
+    "monarch-no-torch @ file:///${PROJECT_ROOT}/assets/wheels/monarch_no_torch-0.1.0.dev20250815-py3-none-any.whl",
 ]
 
 
 [project.scripts]
 forge = "forge.cli.forge:main"
 
-[tool.setuptools.dynamic]
-version = {file = "assets/version.txt"}
-
 
 # ---- Explicit project build information ---- #
 [build-system]
@@ -81,6 +77,10 @@ name = "pytorch-nightly-cu128"
 url  = "https://download.pytorch.org/whl/nightly/cu128"
 #explicit = true
 
+[[tool.uv.index]]
+name = "pytorch-nightly-cpu"
+url = "https://download.pytorch.org/whl/nightly/cpu"
+
 # vllm
 # [[tool.uv.index]]
 # name = "vllm-nightly"
@@ -89,7 +89,7 @@ url  = "https://download.pytorch.org/whl/nightly/cu128"
 
 [tool.uv.sources]
 torchtitan = { index = "pytorch-nightly-cu128" }
-torch = { index = "pytorch-nightly-cu128" }
+torch = { index = "pytorch-nightly-cpu" }
 torchstore = { git = "ssh://[email protected]/meta-pytorch/torchstore.git" }
 #vllm = { index = "vllm-nightly" }