From 599e0e96a7745f035feafa633325554aa5a2b642 Mon Sep 17 00:00:00 2001
From: anton <gan4x4@gmail.com>
Date: Tue, 17 Oct 2023 18:09:57 +0300
Subject: [PATCH] Add external url for legacy pytorch package in
 environment.yml and fix dataset path according to readme.md

---
 dataset.py      | 5 ++---
 environment.yml | 1 +
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/dataset.py b/dataset.py
index acbd24b3..68d027d2 100644
--- a/dataset.py
+++ b/dataset.py
@@ -24,8 +24,7 @@
 class ISIC2016(Dataset):
     def __init__(self, args, data_path , transform = None, transform_msk = None, mode = 'Training',prompt = 'click', plane = False):
 
-
-        df = pd.read_csv(os.path.join(data_path, 'ISBI2016_ISIC_Part3B_' + mode + '_GroundTruth.csv'), encoding='gbk')
+        df = pd.read_csv(os.path.join(data_path, 'ISBI2016_ISIC_Part1_' + mode + '_GroundTruth.csv'), encoding='gbk')
         self.name_list = df.iloc[:,1].tolist()
         self.label_list = df.iloc[:,2].tolist()
         self.data_path = data_path
@@ -91,4 +90,4 @@ def __getitem__(self, index):
             'pt':pt,
             'image_meta_dict':image_meta_dict,
         }
-        
\ No newline at end of file
+        
diff --git a/environment.yml b/environment.yml
index 786e8e30..f2747d7c 100644
--- a/environment.yml
+++ b/environment.yml
@@ -173,6 +173,7 @@ dependencies:
   - zlib=1.2.13=h5eee18b_0
   - zstd=1.5.5=hc292b87_0
   - pip:
+    - --extra-index-url https://download.pytorch.org/whl/cu113
     - aiosignal==1.2.0
     - alembic==1.10.4
     - appdirs==1.4.4