black and ruff for linting and hook

Author: SamoraHunter

.pre-commit-config.yaml

@@ -1,5 +1,18 @@
 repos:
   - repo: https://github.com/kynan/nbstripout
-    rev: master  
+    rev: 0.7.1
     hooks:
       - id: nbstripout
+
+  - repo: https://github.com/psf/black
+    rev: 24.4.2
+    hooks:
+      - id: black
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    # Find the latest version at https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.4.4
+    hooks:
+      # Run the linter and fix autofixable issues (like import sorting).
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]

ml_grid/__init__.py

@@ -1,2 +1,4 @@
 from ml_grid.pipeline import data, read_in
 from ml_grid.util import param_space
+
+__all__ = ["data", "read_in", "param_space"]