From 31cee666dc9372ceb79f415f521fc6293a822b3b Mon Sep 17 00:00:00 2001 From: Antony Milne Date: Thu, 18 Jul 2024 21:37:59 +0100 Subject: [PATCH] Fix lint hopefully --- vizro-core/src/vizro/models/types.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vizro-core/src/vizro/models/types.py b/vizro-core/src/vizro/models/types.py index cfa616724..8dce74d74 100644 --- a/vizro-core/src/vizro/models/types.py +++ b/vizro-core/src/vizro/models/types.py @@ -96,8 +96,9 @@ def __init__(self, function, /, *args, **kwargs): self.__bound_arguments.update(self.__bound_arguments[var_keyword_param]) del self.__bound_arguments[var_keyword_param] - # This is used to check that the mode of the capture decorator matches the inserted captured callable. + # Used in later validations of the captured callable. self._mode = None + self._model_example = None def __call__(self, *args, **kwargs): """Run the `function` with the initially bound arguments overridden by `**kwargs`. @@ -283,7 +284,7 @@ class capture: def __init__(self, mode: Literal["graph", "action", "table", "ag_grid", "figure"]): """Decorator to capture a function call.""" - # mode and model are used in later validations of the captured callable. + # mode and model_example are used in later validations of the captured callable. self._mode = mode model_examples = { "graph": "vm.Graph(figure=...)",