Merge branch 'hotfix/0.3.1' into main

.github/workflows/install.yaml

@@ -0,0 +1,48 @@
+name: Composer Pip Install
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+  workflow_dispatch: {}
+jobs:
+  ComposerPip:
+    timeout-minutes: 10
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python_version:
+          - "3.7"
+          - "3.8"
+          - "3.9"
+        install_version:
+          - ""
+          - "[all]"
+          - "[dev]"
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python_version }}
+      - name: Cache pip
+        uses: actions/cache@v2
+        with:
+          # This path is specific to Ubuntu
+          path: ~/.cache/pip
+          # Look to see if there is a cache hit for the corresponding requirements file
+          key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-
+            ${{ runner.os }}-
+      - name: Setup
+        run: |
+          set -exuo pipefail
+          python -m pip install --upgrade pip wheel
+          python -m pip install -e .${{ matrix.install_version }}
+      - name: Run import
+        id: tests
+        run: |
+          set -exuo pipefail
+          python -c "import composer"

setup.py

@@ -71,7 +71,7 @@ def package_files(directory):
 
 setup(
     name="mosaicml",
-    version="0.3.0",
+    version="0.3.1",
     author="MosaicML",
     author_email="team@mosaicml.com",
     description="composing methods for ML training efficiency",
@@ -83,7 +83,7 @@ def package_files(directory):
         "composer": ['py.typed'],
         "": package_files('composer/yamls'),
     },
-    packages=setuptools.find_packages(include=["composer"]),
+    packages=setuptools.find_packages(exclude=["tests*"]),
     classifiers=[
         "Programming Language :: Python :: 3",
     ],