Defer papermill import in order not to add it to base requirements

Signed-off-by: Pedro Algarvio <[email protected]>

Author: s0undt3ch

changelog/68.improvement.rst

@@ -1,4 +1,4 @@
 Several improvements to the project
 
 * The `salt-analytics.examples` package now uses the same version as `salt-analytics-framework`
-* Defer imports of uncommon usage libraries to reduce base requirements
+* Defer imports of uncommon usage libraries to reduce base requirements(for example, keras, numpy, papermill)

requirements/jupyter.txt

@@ -0,0 +1 @@
+papermill

requirements/tests.txt

@@ -1,5 +1,5 @@
 -r base.txt
+-r jupyter.txt
 pytest>=6.0.0
 pytest-salt-factories==1.0.0rc21
 pytest-asyncio
-papermill

setup.cfg

@@ -74,6 +74,7 @@ extras_require =
   docsauto = requirements/docs-auto.txt
   changelog = requirements/changelog.txt
   build = requirements/build.txt
+  jupyter = requirements/jupyter.txt
 
 
 [bdist_wheel]

src/saf/process/jupyter_notebook.py

@@ -14,8 +14,6 @@
 from typing import Optional
 from typing import Type
 
-import papermill
-
 from saf.models import CollectedEvent
 from saf.models import PipelineRunContext
 from saf.models import ProcessConfigBase
@@ -51,6 +49,8 @@ async def process(
     """
     Run the jupyter notebook, doing papermill parameterizing using the event data given.
     """
+    import papermill
+
     output = ctx.config.output_notebook or ctx.config.notebook
     params = ctx.config.params.copy()
     for key in ctx.config.input_keys: