haliax 1.4.dev299__tar.gz → 1.4.dev300__tar.gz

{haliax-1.4.dev299 → haliax-1.4.dev300}/.github/workflows/run_pre_commit.yaml

@@ -1,6 +1,6 @@
 name: Pre-Commit
 
-on: [push]
+on: [push, pull_request]
 
 jobs:
   build:

haliax-1.4.dev300/.github/workflows/run_quick_levanter_tests.yaml

@@ -0,0 +1,36 @@
+name: Run Levanter Tests
+
+on: [pull_request]
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python 3.10.11
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.10.11
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install flake8 pytest
+          pip install --upgrade "jax[cpu]==0.4.26" "jaxlib[cpu]==0.4.26"
+
+      - name: Install Levanter from source
+        run: |
+          cd ..
+          git clone https://github.com/stanford-crfm/levanter.git
+          cd levanter
+          pip install -e .
+      - name: Install Haliax on top
+        run: |
+          # install second since levanter will install a built version of haliax
+          cd ../haliax
+          pip install .[dev]
+      - name: Test levanter with pytest
+        run: |
+          cd ../levanter
+          XLA_FLAGS=--xla_force_host_platform_device_count=8 PYTHONPATH=tests:src:../src pytest tests -m "not entry and not slow"

{haliax-1.4.dev299 → haliax-1.4.dev300}/.github/workflows/run_tests.yaml

@@ -1,21 +1,18 @@
 name: Run Tests
 
-on: [push]
+on: [push, pull_request]
 
 jobs:
   build:
 
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.10.11"]
 
     steps:
       - uses: actions/checkout@v3
-      - name: Set up Python ${{ matrix.python-version }}
+      - name: Set up Python 3.10.11
         uses: actions/setup-python@v4
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: 3.10.11
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip

{haliax-1.4.dev299 → haliax-1.4.dev300}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: haliax
-Version: 1.4.dev299
+Version: 1.4.dev300
 Summary: Named Tensors for Legible Deep Learning in JAX
 Project-URL: Homepage, https://github.com/stanford-crfm/haliax
 Project-URL: Bug Tracker, https://github.com/stanford-crfm/haliax/issues/

haliax-1.4.dev300/src/haliax/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.4.dev300"

haliax-1.4.dev299/src/haliax/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.4.dev299"