Add jupyter notebooks to ecosystem checks

astral-sh · Dec 28, 2023 · b0ec884 · b0ec884
1 parent 57b6a8c
commit b0ec884
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/python/ruff-ecosystem/ruff_ecosystem/defaults.py b/python/ruff-ecosystem/ruff_ecosystem/defaults.py
@@ -9,6 +9,8 @@
     Repository,
 )
 
+JUPYTER_NOTEBOOK_SELECT = "A,E703,F704,B015,B018,D100"
+
 # TODO(zanieb): Consider exporting this as JSON and loading from there instead
 DEFAULT_TARGETS = [
     Project(repo=Repository(owner="DisnakeDev", name="disnake", ref="master")),
@@ -93,4 +95,15 @@
         repo=Repository(owner="zulip", name="zulip", ref="main"),
         check_options=CheckOptions(select="ALL"),
     ),
+    # Jupyter Notebooks
+    Project(
+        repo=Repository(owner="huggingface", name="notebooks", ref="main"),
+        check_options=CheckOptions(select=JUPYTER_NOTEBOOK_SELECT),
+        config_overrides={"include": ["*.ipynb"]},
+    ),
+    Project(
+        repo=Repository(owner="openai", name="openai-cookbook", ref="main"),
+        check_options=CheckOptions(select=JUPYTER_NOTEBOOK_SELECT),
+        config_overrides={"include": ["*.ipynb"]},
+    ),
 ]