From 7c11034b627b1eec61aade26713ebaeda8c46e40 Mon Sep 17 00:00:00 2001
From: Antonio Mirarchi <antonio.mirarchi@upf.edu>
Date: Mon, 13 May 2024 20:23:15 +0200
Subject: [PATCH] add torch_ema to enviroment.yml

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

diff --git a/environment.yml b/environment.yml
index 697fe8084..fc1eb3225 100644
--- a/environment.yml
+++ b/environment.yml
@@ -17,3 +17,5 @@ dependencies:
   - pytest
   - psutil
   - gxx<12
+  - pip:
+    - torch-ema