chore: fix mypy issues

DHARPA-Project · Dec 19, 2023 · e112d0c · e112d0c
1 parent b187b64
commit e112d0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/kiara_plugin/jupyter/renderers/pipeline.py b/src/kiara_plugin/jupyter/renderers/pipeline.py
@@ -72,12 +72,12 @@ def retrieve_jinja_env(self) -> JinjaEnv:
         jinja_env = JinjaEnv(template_base="kiara_plugin.jupyter")
         return jinja_env
 
-    def get_template(self, render_config: RenderInputsSchema) -> Template:
+    def get_template(self, render_config: JupyterNotebookRenderInputSchema) -> Template:
 
         return self.get_jinja_env().get_template("pipeline/jupyter_notebook.ipynb.j2")
 
     def assemble_render_inputs(
-        self, instance: Any, render_config: RenderInputsSchema
+        self, instance: Any, render_config: JupyterNotebookRenderInputSchema
     ) -> Mapping[str, Any]:
 
         from kiara.utils.rendering import create_pipeline_render_inputs