Defer imports

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

Author: s0undt3ch

changelog/68.improvement.rst

@@ -1,3 +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

examples/src/saltext/safexamples/collect/mnist_digits.py

@@ -12,8 +12,6 @@
 from typing import AsyncIterator
 from typing import Type
 
-from tensorflow import keras
-
 from saf.models import CollectConfigBase
 from saf.models import CollectedEvent
 from saf.models import PipelineRunContext
@@ -41,6 +39,8 @@ async def collect(*, ctx: PipelineRunContext[MNISTDigitsConfig]) -> AsyncIterato
     """
     Periodically yield a random MNIST test digit and it's desired output.
     """
+    from tensorflow import keras
+
     file_path = pathlib.Path(ctx.config.path)
     log.debug("Downloading the MNIST digits dataset to %s", file_path)
     (x_train, y_train), (x_test, y_test) = keras.datasets.mnist.load_data(path=file_path)

examples/src/saltext/safexamples/process/mnist_network.py

@@ -10,9 +10,6 @@
 from typing import AsyncIterator
 from typing import Type
 
-import numpy as np
-from tensorflow import keras
-
 from saf.models import CollectedEvent
 from saf.models import PipelineRunContext
 from saf.models import ProcessConfigBase
@@ -43,6 +40,9 @@ async def process(
     """
     Run the MNIST network.
     """
+    import numpy as np
+    from tensorflow import keras
+
     if "mnist_model" not in ctx.cache:
         model_path = pathlib.Path(ctx.config.model)
         log.debug("Loading the mnist model from %s", model_path)

examples/src/saltext/safexamples/process/numpy_save_keys.py

@@ -10,8 +10,6 @@
 from typing import AsyncIterator
 from typing import Type
 
-import numpy as np
-
 from saf.models import CollectedEvent
 from saf.models import PipelineRunContext
 from saf.models import ProcessConfigBase
@@ -42,6 +40,8 @@ async def process(
     """
     Save the keys using numpy.
     """
+    import numpy as np
+
     config = ctx.config
     if not config.base_path.exists():
         config.base_path.mkdir(parents=True)