From 6b5d1b4cd4d650f2f558ed4ab4035f6a19e2455f Mon Sep 17 00:00:00 2001
From: Francois Caud <frcaud@gmail.com>
Date: Thu, 21 Dec 2023 17:44:50 +0100
Subject: [PATCH] add hugging face transformers and datasets

---
 environment.yml     | 2 ++
 extra_libraries.txt | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/environment.yml b/environment.yml
index d520d49..8312272 100644
--- a/environment.yml
+++ b/environment.yml
@@ -18,6 +18,8 @@ dependencies:
   - lightgbm
   - nilearn
   - scikit-image>=0.18
+  - transformers
+  - datasets
 
   - pip
   - pip:
diff --git a/extra_libraries.txt b/extra_libraries.txt
index c18d77b..64972d6 100644
--- a/extra_libraries.txt
+++ b/extra_libraries.txt
@@ -5,3 +5,5 @@ lightgbm
 xgboost
 catboost
 torchvision
+transformers
+datasets