From b911d3c8d3210593aba0903f36360a38cb045973 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 24 Nov 2023 12:42:22 -0800 Subject: [PATCH] more deps --- .github/workflows/test.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 3f7dae5..243d16d 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -20,8 +20,6 @@ jobs: python-version: 3.11 - name: Install Dependencies run: pip install numpy tqdm - - name: Get code size - run: PYTHONPATH="." python sz.py - name: Train MNIST run: PYTHONPATH="." python mnist.py @@ -37,8 +35,10 @@ jobs: uses: actions/setup-python@v4 with: python-version: 3.11 - - name: Install deps with mypy torch for testing - run: pip install numpy tqdm mypy torch --extra-index-url https://download.pytorch.org/whl/cpu + - name: Install deps with testing deps + run: pip install numpy tqdm mypy torch pytest tabulate --extra-index-url https://download.pytorch.org/whl/cpu + - name: Get code size + run: PYTHONPATH="." python sz.py - name: Test ops dtype optim run: | PYTHONPATH="." python test/test_ops.py