From 25825da7d94013c756105c81841d6b0b220f6a0e Mon Sep 17 00:00:00 2001
From: R-Poulard <remipoul@gmail.com>
Date: Fri, 22 Dec 2023 07:49:10 +0100
Subject: [PATCH] add in environment.yml

---
 environment.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/environment.yml b/environment.yml
index 0ad71d4..5e363fc 100644
--- a/environment.yml
+++ b/environment.yml
@@ -18,7 +18,8 @@ dependencies:
   - tensorflow-cpu
   - pytorch-cpu
   - torchvision-cpu
-  
+  - imbalanced-learn
+
   - pip
   - pip:
     - ramp-workflow==0.6.0