Cleanup to avoid ruff warning

Author: YoanSallami

synalinks/src/callbacks/progbar_logger.py

@@ -127,7 +127,9 @@ def _update_progbar(self, batch, logs=None, finalize=False):
 
         if self.verbose == 1:
             if is_marimo_notebook_running():
-                self.progbar.progress.update(increment=self.seen, subtitle=format_logs(logs))
+                self.progbar.progress.update(
+                    increment=self.seen, subtitle=format_logs(logs)
+                )
                 if finalize:
                     self.progbar.disabled = True
             else:

synalinks/src/metrics/reduction_metrics.py

@@ -105,9 +105,7 @@ def __init__(self, name="mean", in_mask=None, out_mask=None):
     async def update_state(self, values):
         values = reduce_to_samplewise_values(values, reduce_fn=numpy.mean)
         total = self.total_with_count.get("total")
-        self.total_with_count.update(
-            {"total": float(total + numpy.sum(values))}
-        )
+        self.total_with_count.update({"total": float(total + numpy.sum(values))})
         if len(values.shape) >= 1:
             num_samples = numpy.shape(values)[0]
         else:

synalinks/src/modules/agents/react.py

@@ -119,9 +119,9 @@ async def calculate(expression: str):
             the decision prompt (Default to False) (see `Decision`).
         use_outputs_schema (bool): Optional. Whether or not use the outputs schema in
             the decision prompt (Default to False) (see `Decision`).
-        return_inputs_with_trajectory (bool): Optional. Whether or not to concatenate the inputs along 
-            with the agent trajectory to the outputs (Default to False).
-        return_inputs_only (bool): Optional. Whether or not to concatenate the inputs 
+        return_inputs_with_trajectory (bool): Optional. Whether or not to concatenate the
+            inputs along with the agent trajectory to the outputs (Default to False).
+        return_inputs_only (bool): Optional. Whether or not to concatenate the inputs
             to the outputs (Default to False).
         max_iterations (int): The maximum number of steps to perform.
         name (str): Optional. The name of the module.
@@ -258,10 +258,10 @@ async def build(self, inputs):
                     finish_branches.append(last_step)
 
         final = await Or()(finish_branches)
-        
+
         if self.return_inputs_with_trajectory:
             final.factorize()
-            
+
         if self.return_inputs_only:
             final = inputs + final
 

synalinks/src/ops/json.py

@@ -238,7 +238,7 @@ async def logical_or(x1, x2, name=None, description=None):
         name=name,
         description=description,
     )(x1, x2)
-    
+
 
 class Xor(Operation):
     """Perform a logical `Xor` operation between data models."""

synalinks/src/ops/json_test.py

@@ -11,10 +11,10 @@
 from synalinks.src.ops.json import concat
 from synalinks.src.ops.json import factorize
 from synalinks.src.ops.json import in_mask
-from synalinks.src.ops.json import out_mask
 from synalinks.src.ops.json import logical_and
 from synalinks.src.ops.json import logical_or
 from synalinks.src.ops.json import logical_xor
+from synalinks.src.ops.json import out_mask
 from synalinks.src.programs import Program
 from synalinks.src.utils.nlp_utils import remove_numerical_suffix
 
@@ -179,27 +179,27 @@ class Test(DataModel):
             for var2 in new_program.variables:
                 if remove_numerical_suffix(var2.path) == var1.path:
                     self.assertEqual(var2.get_json(), var1.get_json())
-                    
+
 
 class LogicalAndTest(testing.TestCase):
     async def test_and_table(self):
         class Answer(DataModel):
             answer: str
-            
+
         class Result(DataModel):
             answer: str
             answer_1: str
-        
+
         i0 = Input(data_model=Answer)
         i1 = Input(data_model=Answer)
-        
+
         x0 = await logical_and(i0, i1)
-        
+
         program = Program(
             inputs=[i0, i1],
             outputs=x0,
         )
-        
+
         result = await program([Answer(answer="Paris"), Answer(answer="Toulouse")])
         self.assertEqual(
             result.get_json(),
@@ -208,36 +208,36 @@ class Result(DataModel):
                 answer_1="Toulouse",
             ).get_json(),
         )
-        
+
         result = await program([Answer(answer="Paris"), None])
         self.assertEqual(result, None)
-        
+
         result = await program([None, Answer(answer="Toulouse")])
         self.assertEqual(result, None)
-        
+
         result = await program([None, None])
         self.assertEqual(result, None)
-        
+
 
 class LogicalOrTest(testing.TestCase):
     async def test_or_table(self):
         class Answer(DataModel):
             answer: str
-            
+
         class Result(DataModel):
             answer: str
             answer_1: str
-        
+
         i0 = Input(data_model=Answer)
         i1 = Input(data_model=Answer)
-        
+
         x0 = await logical_or(i0, i1)
-        
+
         program = Program(
             inputs=[i0, i1],
             outputs=x0,
         )
-        
+
         result = await program([Answer(answer="Paris"), Answer(answer="Toulouse")])
         self.assertEqual(
             result.get_json(),
@@ -246,45 +246,45 @@ class Result(DataModel):
                 answer_1="Toulouse",
             ).get_json(),
         )
-        
+
         result = await program([Answer(answer="Paris"), None])
         self.assertEqual(result.get_json(), Answer(answer="Paris").get_json())
-        
+
         result = await program([None, Answer(answer="Toulouse")])
         self.assertEqual(result.get_json(), Answer(answer="Toulouse").get_json())
-        
+
         result = await program([None, None])
         self.assertEqual(result, None)
-        
-        
+
+
 class LogicalXorTest(testing.TestCase):
     async def test_xor_table(self):
         class Answer(DataModel):
             answer: str
-            
+
         class Result(DataModel):
             answer: str
             answer_1: str
-        
+
         i0 = Input(data_model=Answer)
         i1 = Input(data_model=Answer)
-        
+
         x0 = await logical_xor(i0, i1)
-        
+
         program = Program(
             inputs=[i0, i1],
             outputs=x0,
         )
-        
+
         result = await program([Answer(answer="Paris"), Answer(answer="Toulouse")])
         self.assertEqual(result, None)
-        
+
         result = await program([Answer(answer="Paris"), None])
         self.assertEqual(result.get_json(), Answer(answer="Paris").get_json())
-        
+
         result = await program([None, Answer(answer="Toulouse")])
         self.assertEqual(result.get_json(), Answer(answer="Toulouse").get_json())
-        
+
         result = await program([None, None])
         self.assertEqual(result, None)