Add dependencies to pyproject and uv.lock (#6)

* Add dependencies

Signed-off-by: Charlie Truong <[email protected]>

* Set tool.uv managed = true

Signed-off-by: Charlie Truong <[email protected]>

* Move no-install-package to docker container

Signed-off-by: Charlie Truong <[email protected]>

* Update uv lock file

Signed-off-by: Charlie Truong <[email protected]>

* Remove torchvision from the uv --no-install-package

Signed-off-by: Charlie Truong <[email protected]>

---------

Signed-off-by: Charlie Truong <[email protected]>

Author: chtruong814

docker/Dockerfile.ci

@@ -31,5 +31,20 @@ RUN --mount=type=bind,source=pyproject.toml,target=/workspace/pyproject.toml \
 
     # Use the container's torch installation rather than reinstall it
     uv venv --system-site-packages ${UV_PROJECT_ENVIRONMENT}
-    uv sync --link-mode symlink --locked --only-group test --only-group dev
+    uv sync --link-mode symlink --locked --all-groups \
+        --no-install-package absl-py \
+        --no-install-package torch \
+        --no-install-package triton \
+        --no-install-package nvidia-cublas-cu12 \
+        --no-install-package nvidia-cuda-cupti-cu12 \
+        --no-install-package nvidia-cuda-nvrtc-cu12 \
+        --no-install-package nvidia-cuda-runtime-cu12 \
+        --no-install-package nvidia-cudnn-cu12 \
+        --no-install-package nvidia-cufft-cu12 \
+        --no-install-package nvidia-cufile-cu12 \
+        --no-install-package nvidia-curand-cu12 \
+        --no-install-package nvidia-cusolver-cu12 \
+        --no-install-package nvidia-cusparse-cu12 \
+        --no-install-package nvidia-cusparselt-cu12 \
+        --no-install-package nvidia-nccl-cu12
 EOF

pyproject.toml

@@ -54,7 +54,11 @@ classifiers = [
     "Topic :: Software Development :: Libraries",
     "Topic :: Utilities",
 ]
-dependencies = []
+dependencies = [
+    "torch",
+    "absl-py",
+    "megatron-core>=0.14.0a0,<0.16.0",
+]
 
 [build-system]
 requires = ["setuptools>=61"]
@@ -83,6 +87,9 @@ dev = [
     "mypy>=1.8.0",
 ]
 
+[tool.uv]
+managed = true
+
 [tool.setuptools]
 packages = ["emerging_optimizers"]