Disable pydantic validation for plot

Author: wwwillchen

mesop/component_helpers/helper.py

@@ -86,40 +86,43 @@ def wrapper(*args: Any, **kwargs: Any):
 C = TypeVar("C", bound=Callable[..., Any])
 
 
-def component(fn: C) -> C:
-  """Wraps a Python function to make it a user-defined component."""
-
-  validated_fn = validate(fn)
-
-  @wraps(fn)
-  def wrapper(*args: Any, **kw_args: Any):
-    prev_current_node = runtime().context().current_node()
-    component = prev_current_node.children.add()
-    source_code_location = None
-    if runtime().debug_mode:
-      source_code_location = get_caller_source_code_location(levels=2)
-    component.MergeFrom(
-      create_component(
-        component_name=get_component_name(fn),
-        proto=pb.UserDefinedType(
-          args=[
-            pb.UserDefinedType.Arg(
-              arg_name=kw_arg, code_value=map_code_value(value)
-            )
-            for kw_arg, value in kw_args.items()
-            if map_code_value(value) is not None
-          ]
-        ),
-        source_code_location=source_code_location,
+def component(skip_validation: bool = False):
+  def component_wrapper(fn: C) -> C:
+    """Wraps a Python function to make it a user-defined component."""
+
+    validated_fn = fn if skip_validation else validate(fn)
+
+    @wraps(fn)
+    def wrapper(*args: Any, **kw_args: Any):
+      prev_current_node = runtime().context().current_node()
+      component = prev_current_node.children.add()
+      source_code_location = None
+      if runtime().debug_mode:
+        source_code_location = get_caller_source_code_location(levels=2)
+      component.MergeFrom(
+        create_component(
+          component_name=get_component_name(fn),
+          proto=pb.UserDefinedType(
+            args=[
+              pb.UserDefinedType.Arg(
+                arg_name=kw_arg, code_value=map_code_value(value)
+              )
+              for kw_arg, value in kw_args.items()
+              if map_code_value(value) is not None
+            ]
+          ),
+          source_code_location=source_code_location,
+        )
       )
-    )
-    runtime().context().set_current_node(component)
-    ret = validated_fn(*args, **kw_args)
-    runtime().context().set_current_node(prev_current_node)
-    return ret
+      runtime().context().set_current_node(component)
+      ret = validated_fn(*args, **kw_args)
+      runtime().context().set_current_node(prev_current_node)
+      return ret
 
-  runtime().register_native_component_fn(fn)
-  return cast(C, wrapper)
+    runtime().register_native_component_fn(fn)
+    return cast(C, wrapper)
+
+  return component_wrapper
 
 
 def get_component_name(fn: Callable[..., Any]) -> pb.ComponentName:

mesop/components/button/button.py

@@ -11,7 +11,7 @@
 from mesop.events import ClickEvent
 
 
-@component
+@component()
 def button(
   label: str | None = None,
   *,

mesop/components/checkbox/checkbox.py

@@ -55,7 +55,7 @@ class CheckboxIndeterminateChangeEvent(MesopEvent):
 )
 
 
-@component
+@component()
 def checkbox(
   label: str | None = None,
   *,

mesop/components/plot/plot.py

@@ -15,7 +15,8 @@ def savefig(self, fname: BytesIO, *, format: str):
     pass
 
 
-@component
+# Skip pydantic validation because Figure is a Protocol which can't type-check properly
+@component(skip_validation=True)
 def plot(figure: Figure, *, style: Style | None = None):
   """
   Creates a plot component from a Matplotlib figure.

mesop/labs/layout.py

@@ -2,6 +2,6 @@
 from mesop.components.box.box import box
 
 
-@component
+@component()
 def columns(columns: int = 2):
   return box(style=Style(columns=columns))