Adds ruff config

Author: JyotinderSingh

.github/workflows/lint.yaml

@@ -0,0 +1,19 @@
+name: Lint
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  ruff:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: astral-sh/ruff-action@v3
+        with:
+          args: check
+      - uses: astral-sh/ruff-action@v3
+        with:
+          args: format --check

.pre-commit-config.yaml

@@ -0,0 +1,7 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.11.12
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format

README.md

@@ -401,6 +401,30 @@ destroy Pulumi-managed resources.
 
 Contributions are welcome. Please read our contributing guidelines before submitting pull requests.
 
+### Development Setup
+
+1. Install the package with dev dependencies:
+
+```bash
+pip install -e ".[dev]"
+```
+
+2. Install pre-commit hooks:
+
+```bash
+pre-commit install
+```
+
+This enables automatic linting and formatting checks (via [Ruff](https://docs.astral.sh/ruff/)) on every commit.
+
+To run the checks manually against all files:
+
+```bash
+pre-commit run --all-files
+```
+
+### Submitting Changes
+
 1. Fork the repository
 2. Create a feature branch (`git checkout -b feature/amazing-feature`)
 3. Commit your changes (`git commit -m 'Add amazing feature'`)

pyproject.toml

@@ -14,6 +14,7 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
+    "absl-py",
     "cloudpickle",
     "numpy",
     "keras",
@@ -32,6 +33,10 @@ cli = [
     "pulumi-command>=1.0",
     "google-cloud-tpu>=1.0",
 ]
+dev = [
+    "pre-commit",
+    "ruff",
+]
 demo = [
     "jax",
     "keras",
@@ -53,3 +58,11 @@ requires = [
     "twine",
 ]
 build-backend = "hatchling.build"
+
+[tool.ruff]
+indent-width = 2
+line-length = 80
+
+[tool.ruff.lint]
+select = ["B", "E", "F", "N", "PYI", "T20", "TID", "SIM", "W", "I", "NPY"]
+ignore = ["E501"]