chore(dev): use ruff as linter and formatter

.github/workflows/full-tests.yml

@@ -16,7 +16,7 @@ jobs:
     strategy:
       matrix:
         os: [Windows, macOS] # exclude Ubuntu as it is available in pr-tests
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
         include:
           - os: Windows
             image: windows-2022

.github/workflows/latest-deps-tests.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       matrix:
         os: [Ubuntu]
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
         include:
           - os: Ubuntu
             image: ubuntu-latest

.github/workflows/pr-tests.yml

@@ -11,7 +11,7 @@ jobs:
     strategy:
       matrix:
         os: [Ubuntu]
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
         include:
           - os: Ubuntu
             image: ubuntu-latest

.github/workflows/test-and-build-wheel.yml

@@ -86,7 +86,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
     steps:
       - name: Set up Python
         uses: actions/setup-python@v5

.github/workflows/test-published-dist.yml

@@ -9,7 +9,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
     steps:
       - name: Set up Python
         uses: actions/setup-python@v5

.pre-commit-config.yaml

@@ -1,20 +1,20 @@
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.4.0
+
     hooks:
       - id: check-yaml
       - id: end-of-file-fixer
       - id: trailing-whitespace
-  - repo: https://github.com/pycqa/isort
-    rev: 5.12.0
-    hooks:
-      - id: isort
-        args: [ "--profile", "black" ]
-        name: isort (python)
-  - repo: https://github.com/psf/black
-    rev: 23.3.0
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.9.6
+
     hooks:
-      - id: black
+      - id: ruff
+        args: ["--fix"]
+      - id: ruff-format
+
   - repo: https://github.com/Lucas-C/pre-commit-hooks
     rev: v1.4.2
     hooks:
@@ -23,7 +23,6 @@ repos:
         args:
           - --license-filepath
           - LICENSE.md
-
 # Deactivating this for now.
 #  - repo: https://github.com/pycqa/pylint
 #    rev: v2.17.0

build_notebook_docs.py

@@ -87,7 +87,7 @@ def _fix_prefix_and_type_in_code_blocks(md_file_path):
                     updated_block = "\n".join(lines)
                     content = content.replace(block, updated_block)
                     block = updated_block
-            except:
+            except yaml.YAMLError:
                 pass
 
         if lines[0] == "```" and "from nemoguardrails" in block:
@@ -194,9 +194,7 @@ def rename_md_to_readme(start_dir):
 
         # We do some additional post-processing
         _remove_code_blocks_with_text(readme_path.absolute(), "# Init:")
-        _remove_code_blocks_with_text(
-            readme_path.absolute(), "# Hide from documentation page."
-        )
+        _remove_code_blocks_with_text(readme_path.absolute(), "# Hide from documentation page.")
 
         _remove_code_blocks_with_text(
             readme_path.absolute(),

docs/colang-2/examples/test_csl.py

@@ -17,12 +17,12 @@
 import sys
 
 import pytest
+from utils import compare_interaction_with_test_script
 
 pathlib.Path(__file__).parent.parent.parent.resolve()
 sys.path.append(str(pathlib.Path(__file__).parent.parent.parent.parent.resolve()))
 print(sys.path)
 
-from utils import compare_interaction_with_test_script
 
 ########################################################################################################################
 # CORE
@@ -637,9 +637,7 @@ async def test_repeating_timer():
 # USAGE_END: test_repeating_timer
         """
 
-    await compare_interaction_with_test_script(
-        test_script, colang_code, wait_time_s=2.0
-    )
+    await compare_interaction_with_test_script(test_script, colang_code, wait_time_s=2.0)
 
 
 @pytest.mark.asyncio
@@ -809,9 +807,7 @@ async def test_polling_llm_request_response():
 # USAGE_END: test_polling_llm_request_response
         """
 
-    await compare_interaction_with_test_script(
-        test_script, colang_code, llm_responses=['"nine"']
-    )
+    await compare_interaction_with_test_script(test_script, colang_code, llm_responses=['"nine"'])
 
 
 @pytest.mark.asyncio

docs/colang-2/examples/utils.py

@@ -92,6 +92,6 @@ async def compare_interaction_with_test_script(
     )
     clean_test_script = cleanup(test_script)
     clean_result = cleanup(result)
-    assert (
-        clean_test_script == clean_result
-    ), f"\n----\n{clean_result}\n----\n\ndoes not match test script\n\n----\n{clean_test_script}\n----"
+    assert clean_test_script == clean_result, (
+        f"\n----\n{clean_result}\n----\n\ndoes not match test script\n\n----\n{clean_test_script}\n----"
+    )