Merge pull request #5 from pytorch-labs/ci1

Test installation CI

Author: cpuhrsch

.github/workflows/test_install.yml

@@ -0,0 +1,31 @@
+name: Test Installation
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  test-installation:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v2
+
+    - name: Set up Python 3.8
+      uses: actions/setup-python@v2
+      with:
+        python-version: 3.8
+
+    - name: Install dependencies
+      run: |
+        pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
+        pip install -e .
+
+    - name: Test import
+      run: |
+        python -c "from torchao import quantization"

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name='torchao',
-    version='0.1',
+    version='0.0.1',
     packages=find_packages(),
     install_requires=[
         'torch',