Add linting via pre-commit (#2)

Author: jakevdp

.github/workflows/lint.yaml

@@ -0,0 +1,24 @@
+name: Lint
+
+on:
+  # Trigger the workflow on push or pull request, but only on main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+permissions:
+  contents: read  # to fetch code
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python 3.12
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.12
+      - uses: pre-commit/action@v3.0.0

.gitignore

@@ -0,0 +1,2 @@
+*.egg-info
+Pipfile

.pre-commit-config.yaml

@@ -0,0 +1,26 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.6.0
+    hooks:
+      - id: check-ast
+      - id: check-merge-conflict
+      - id: check-toml
+      - id: check-yaml
+      - id: end-of-file-fixer
+      - id: trailing-whitespace
+      - id: debug-statements
+  - repo: https://github.com/google/pyink
+    rev: 24.3.0
+    hooks:
+      - id: pyink
+        language_version: python3.12
+        args: [
+          "--line-length=80",
+          "--preview",
+          "--pyink-indentation=2",
+          "--pyink-use-majority-quotes"
+        ]
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.6.1
+    hooks:
+    - id: ruff

LICENSE

@@ -200,4 +200,3 @@
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
-   

README.md

@@ -3,4 +3,4 @@
 This is a work-in-progress meta-package for the AI/ML stack built on top of the
 [jax](http://github.com/google/jax/) package. It is intended as a location for
 tests, documentation, and installation instructions that cover multiple
-packages in the JAX ecosystem.
+packages in the JAX ecosystem.

jax_ml_stack/__init__.py

@@ -1,2 +1,3 @@
 """jax_ml_stack metapackage."""
+
 __version__ = "0.0.1"  # keep in sync with pyproject.toml