From a3c0dae6588ac4bbaeacabceb8602c3826ef55f2 Mon Sep 17 00:00:00 2001
From: ggmarshall <george.marshall.20@ucl.ac.uk>
Date: Mon, 21 Oct 2024 23:18:39 +0200
Subject: [PATCH] add ml packages

---
 templates/config.json | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/templates/config.json b/templates/config.json
index 9fd0d0f..0d1320d 100644
--- a/templates/config.json
+++ b/templates/config.json
@@ -59,7 +59,10 @@
         "pylegendmeta": "pylegendmeta==0.10.2",
         "dspeed": "dspeed==1.6.1",
         "legend-pydataobj": "legend-pydataobj==1.7.0",
-        "legend-daq2lh5": "legend-daq2lh5==1.2.1"
+        "legend-daq2lh5": "legend-daq2lh5==1.2.1",
+        "tensorflow": "tensorflow==2.17",
+        "keras": "keras==3.6.0",
+        "jax": "jax==0.4.30"
       }
     }
   }