feat(language/python): add ipynb package

thm-mni-ii · Nov 4, 2024 · 28cdd7c · 28cdd7c
1 parent c42522e
commit 28cdd7c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/language/python/Earthfile b/language/python/Earthfile
@@ -5,6 +5,7 @@ SETUP:
     ARG NAMESPACE=base
     ARG VERSION=3.11
     ARG ADDITIONAL_PACKAGES
+    ARG ADDITIONAL_PIP_PACKAGES
     ARG COMPUTE_SPECIFIC_CHANNELS
     ARG COMPUTE_SPECIFIC_PACKAGES
     RUN micromamba install -n $NAMESPACE -c pytorch -c conda-forge $COMPUTE_SPECIFIC_CHANNELS -y \
@@ -22,6 +23,7 @@ SETUP:
     mlxtend jax \
     networkx ortools-python \
     $ADDITIONAL_PACKAGES
+    RUN micromamba run -n $NAMESPACE pip install ipynb $ADDITIONAL_PIP_PACKAGES
 
 SETUP_CPU:
     FUNCTION