From 875f2eaf2676e3e5c7ee430a5d72c00b5b66f9b6 Mon Sep 17 00:00:00 2001 From: Advait Jain Date: Mon, 13 May 2024 21:52:01 -0700 Subject: [PATCH] Added instructions for how to install the nightly package. (#3) * Added instructions for how to install the nightly package. BUG=http://b/340367182 * update workflow to test the nightly. --- .github/workflows/nightly_pip_test.yml | 4 ++-- README.md | 2 +- setup.py | 4 +++- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/.github/workflows/nightly_pip_test.yml b/.github/workflows/nightly_pip_test.yml index dcc8bb61..5d88ec1d 100644 --- a/.github/workflows/nightly_pip_test.yml +++ b/.github/workflows/nightly_pip_test.yml @@ -15,7 +15,7 @@ jobs: strategy: matrix: python-version: ["3.9", "3.10", "3.11"] - + runs-on: ubuntu-latest steps: - uses: actions/setup-python@v4 @@ -26,7 +26,7 @@ jobs: pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt - name: Install ai-edge-torch run: | - pip install ai-edge-torch + pip install ai-edge-torch-nightly - name: Import ai-edge-torch run: | python -c "import ai_edge_torch" diff --git a/README.md b/README.md index 7fe5d7f1..b656e064 100644 --- a/README.md +++ b/README.md @@ -101,7 +101,7 @@ pip install ai-edge-torch==0.1.1 Alternately, the nightly version can be installed with: ```bash pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt -pip install ai-edge-torch +pip install ai-edge-torch-nightly ``` * The list of versioned releases can be seen [here](https://github.com/google-ai-edge/ai-edge-torch/releases). diff --git a/setup.py b/setup.py index 0b9ca240..2b8496d9 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,9 @@ then be run with TensorFlow Lite and MediaPipe. This enables applications for Android, iOS and IOT that can run models completely on-device. -More details are in the project's [GitHub repository](https://github.com/google-ai-edge/ai-edge-torch). +[Install steps](https://github.com/google-ai-edge/ai-edge-torch#installation) +and additional details are in the AI Edge Torch +[GitHub repository](https://github.com/google-ai-edge/ai-edge-torch). """.lstrip() setup(