Add ruff rules for flake8-simplify (#549)

Author: cbornet

libs/knowledge-graph/ragstack_knowledge_graph/render.py

@@ -36,7 +36,7 @@ def render_graph_documents(
 
     def _node_id(node: Node) -> int:
         node_key = (node.id, node.type)
-        if node_id := nodes.get(node_key, None):
+        if node_id := nodes.get(node_key):
             return node_id
         else:
             node_id = f"{len(nodes)}"

libs/ragulate/ragstack_ragulate/analysis.py

@@ -89,8 +89,7 @@ def output_box_plots_by_dataset(self, df: DataFrame, metrics: List[str]):
 
         for dataset in datasets:
             fig = go.Figure()
-            test_index = 0
-            for recipe in recipes:
+            for test_index, recipe in enumerate(recipes):
                 y = []
                 x = []
                 q1 = []
@@ -124,7 +123,6 @@ def output_box_plots_by_dataset(self, df: DataFrame, metrics: List[str]):
                         boxpoints=False,  # Do not show individual points
                     )
                 )
-                test_index += 1
 
             fig.update_traces(
                 orientation="h",

libs/ragulate/ragstack_ragulate/cli_commands/run.py

@@ -80,7 +80,7 @@ def call_run(config_file: str, **kwargs):
     for query_pipeline in query_pipelines:
         query_pipeline.query()
 
-    recipe_names = [n for n in config.recipes.keys()]
+    recipe_names = config.recipes.keys()
 
     analysis = Analysis()
     analysis.compare(recipes=recipe_names)

pyproject.toml

@@ -67,10 +67,10 @@ select = [
     "F",
     "FLY",
     "I",
-    "ISC",
     "N",
     "NPY",
     "PERF",
+    "SIM",
     "UP",
     "W",
 ]