From 5ad9f9553c8b45aca14983fad4230e43cbfc3f71 Mon Sep 17 00:00:00 2001
From: Soraya Arias <soraya.arias@inria.fr>
Date: Mon, 30 Dec 2024 19:53:33 +0100
Subject: [PATCH] Add missing packages

---
 docker/requirements-cpu.txt | 2 ++
 docker/requirements-gpu.txt | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/docker/requirements-cpu.txt b/docker/requirements-cpu.txt
index f080543..5b9eab0 100644
--- a/docker/requirements-cpu.txt
+++ b/docker/requirements-cpu.txt
@@ -28,6 +28,8 @@
  Matplotlib
  plotly
  seaborn
+ einops
+ datasets
  barviz
  pyarrow
  Pandas
diff --git a/docker/requirements-gpu.txt b/docker/requirements-gpu.txt
index fe31bda..91e964c 100644
--- a/docker/requirements-gpu.txt
+++ b/docker/requirements-gpu.txt
@@ -27,6 +27,8 @@
  Matplotlib
  plotly
  seaborn
+ einops
+ datasets
  barviz
  pyarrow
  Pandas
-- 
GitLab