From 43503d5bb753bc0b1e066c30fbac5a6377c7ec4f Mon Sep 17 00:00:00 2001
From: Jonas-Ian Kuche <jonas-ian.kuche@mni.thm.de>
Date: Mon, 16 Sep 2024 12:42:16 +0200
Subject: [PATCH] feat: add keras mlxtend and jax

---
 language/python/Earthfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/language/python/Earthfile b/language/python/Earthfile
index 644f490..d00a6f7 100644
--- a/language/python/Earthfile
+++ b/language/python/Earthfile
@@ -9,6 +9,7 @@ SETUP:
     ARG COMPUTE_SPECIFIC_PACKAGES
     RUN micromamba install -n $NAMESPACE -c pytorch -c conda-forge $COMPUTE_SPECIFIC_CHANNELS -y \
     python=$VERSION pytorch torchvision torchaudio $COMPUTE_SPECIFIC_PACKAGES \
+    tensorflow keras \
     lightning torchinfo tensorboard tensorboardx \
     transformers diffusers datasets accelerate evaluate optimum \
     sentence-transformers einops \
@@ -18,6 +19,7 @@ SETUP:
     gradio streamlit \
     d2l beautifulsoup4 langdetect spacy mastodon.py \
     fastai::opencv-python-headless \
+    mlxtend jax \
     $ADDITIONAL_PACKAGES
 
 SETUP_CPU: