Fix program test

Author: YoanSallami

notebooks/conversational_applications.py

@@ -8,7 +8,7 @@
 def _():
     import marimo as mo
     import synalinks
-    
+
     synalinks.backend.clear_session()
     return mo, synalinks
 
@@ -57,7 +57,7 @@ async def _(synalinks):
     _x1 = await synalinks.Generator(
         language_model=language_model,
         prompt_template=synalinks.chat_prompt_template(),
-        streaming=False, # Marimo chat don't handle streaming yet
+        streaming=False,  # Marimo chat don't handle streaming yet
     )(_x0)
 
     program = synalinks.Program(

notebooks/first_steps.py

@@ -8,9 +8,9 @@
 def _():
     import marimo as mo
     import synalinks
-    
+
     synalinks.backend.clear_session()
-    
+
     return mo, synalinks
 
 

notebooks/implementing_custom_modules_and_programs_via_subclassing.py

@@ -62,14 +62,10 @@ def _(synalinks):
     from synalinks import ops
 
     class Thinking(synalinks.DataModel):
-        thinking: str = synalinks.Field(
-            description="Your step by step thinking process"
-        )
+        thinking: str = synalinks.Field(description="Your step by step thinking process")
 
     class CritiqueWithReward(synalinks.DataModel):
-        critique: str = synalinks.Field(
-            description="The step by step critique"
-        )
+        critique: str = synalinks.Field(description="The step by step critique")
         reward: float = synalinks.Field(
             description="The reward corresponding to the critique between [0.0, 1.0]",
             le=1.0,
@@ -108,25 +104,23 @@ def __init__(
             self.stop_threshold = stop_threshold
             self.max_iterations = max_iterations
             self.critique_program = critique_program
-            self.prompt_template= prompt_template
+            self.prompt_template = prompt_template
             self.examples = examples
             self.hints = hints
             self.use_inputs_schema = use_inputs_schema
             self.use_outputs_schema = use_outputs_schema
             if not self.critique_program:
                 # If no critique program is provided
                 # We compute the reward in the thinking step
-                thinking_data_model = \
-                    Thinking \
-                    + synalinks.SymbolicDataModel(
-                        schema=self.schema
-                    ) + CritiqueWithReward
+                thinking_data_model = (
+                    Thinking
+                    + synalinks.SymbolicDataModel(schema=self.schema)
+                    + CritiqueWithReward
+                )
             else:
-                thinking_data_model = \
-                    Thinking \
-                    + synalinks.SymbolicDataModel(
-                        schema=self.schema
-                    )
+                thinking_data_model = Thinking + synalinks.SymbolicDataModel(
+                    schema=self.schema
+                )
             # This is for generating the intermediary steps
             self.thinking = synalinks.Generator(
                 data_model=thinking_data_model,
@@ -136,7 +130,7 @@ def __init__(
                 hints=self.hints,
                 use_inputs_schema=self.use_inputs_schema,
                 use_outputs_schema=self.use_outputs_schema,
-                name=self.name+"_thinking_generator",
+                name=self.name + "_thinking_generator",
             )
             # This is going to be the final generator
             self.generator = synalinks.Generator(
@@ -147,7 +141,7 @@ def __init__(
                 hints=self.hints,
                 use_inputs_schema=self.use_inputs_schema,
                 use_outputs_schema=self.use_outputs_schema,
-                name=self.name+"_generator",
+                name=self.name + "_generator",
             )
 
         async def call(self, inputs, training=False):
@@ -167,9 +161,7 @@ async def call(self, inputs, training=False):
                     if reward > self.stop_threshold:
                         break
                     inputs = await ops.concat(
-                        inputs,
-                        thinking,
-                        name=self.name+f"_thinking_{i}"
+                        inputs, thinking, name=self.name + f"_thinking_{i}"
                     )
             return await self.generator(inputs)
 
@@ -190,7 +182,7 @@ def get_config(self):
                 "name": self.name,
                 "description": self.description,
                 "trainable": self.trainable,
-            }        
+            }
             language_model_config = {
                 "language_model": synalinks.saving.serialize_synalinks_object(
                     self.language_model,
@@ -224,6 +216,7 @@ def from_config(cls, config):
                 critique_program=critique_program,
                 **config,
             )
+
     return BacktrackingOfThought, CritiqueWithReward, Thinking, ops
 
 

notebooks/rewards_metrics_and_optimizers.py

@@ -8,9 +8,9 @@
 def _():
     import marimo as mo
     import synalinks
-    
+
     synalinks.backend.clear_session()
-    
+
     return mo, synalinks
 
 
@@ -101,7 +101,6 @@ def _(mo):
 
 @app.cell
 async def _(synalinks):
-
     synalinks.backend.clear_session()
 
     class Query(synalinks.DataModel):

synalinks/src/backend/common/dynamic_json_schema_utils.py

@@ -1,5 +1,6 @@
 # License Apache 2.0: (c) 2025 Yoan Sallami (Synalinks Team)
 
+
 def dynamic_enum(schema, prop_to_update, labels):
     """Update a schema with dynamic Enum string
 

synalinks/src/backend/pydantic/module.py

@@ -1,4 +1,5 @@
 # License Apache 2.0: (c) 2025 Yoan Sallami (Synalinks Team)
 
+
 class PydanticModule:
     pass

synalinks/src/programs/functional.py

@@ -2,6 +2,7 @@
 # Original authors: François Chollet et al. (Keras Team)
 # License Apache 2.0: (c) 2025 Yoan Sallami (Synalinks Team)
 
+import asyncio
 import copy
 import inspect
 import typing

synalinks/src/programs/program_test.py

@@ -46,7 +46,7 @@ class AnswerWithRationale(DataModel):
                     "variable": {
                         "prompt_template": default_prompt_template(),
                         "examples": [],
-                        "hints": default_hints(),
+                        "hints": [],
                         "predictions": [],
                     }
                 }

synalinks/src/trainers/data_adapters/data_adapter.py

@@ -2,6 +2,7 @@
 # Original authors: François Chollet et al. (Keras Team)
 # License Apache 2.0: (c) 2025 Yoan Sallami (Synalinks Team)
 
+
 class DataAdapter:
     """Base class for input data adapters.
 

synalinks/src/utils/plot_history.py

@@ -78,7 +78,7 @@ def plot_history(
         pass
     try:
         import marimo as mo
-        
+
         if mo.running_in_notebook():
             return mo.image(src=to_file).center()
     except ImportError: