From e93e1e46c4564532a5567fc5b206eb8d2b42e3fc Mon Sep 17 00:00:00 2001
From: Dev35 <38446179+Savager3435@users.noreply.github.com>
Date: Tue, 19 Jul 2022 22:51:52 -0600
Subject: [PATCH] Update environment.yaml

fixes ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' that occurs  with previous environment.yaml set up
---
 environment.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/environment.yaml b/environment.yaml
index f36b0e17d..dd6f7b2af 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -13,6 +13,7 @@ dependencies:
     - albumentations==0.4.3
     - opencv-python==4.1.2.30
     - pudb==2019.2
+    - torchmetrics==v0.6.0
     - imageio==2.9.0
     - imageio-ffmpeg==0.4.2
     - pytorch-lightning==1.4.2
@@ -24,4 +25,4 @@ dependencies:
     - transformers==4.3.1
     - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
     - -e git+https://github.com/openai/CLIP.git@main#egg=clip
-    - -e .
\ No newline at end of file
+    - -e .