quick review

microsoft · Oct 16, 2024 · a66c9a7 · a66c9a7
1 parent 3ef9d7e
commit a66c9a7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 13 deletions.
diff --git a/tests/data/glacier_calving_front/data.py b/tests/data/glacier_calving_front/data.py
@@ -1,3 +1,8 @@
+#!/usr/bin/env python3
+
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
 import hashlib
 import os
 import shutil

diff --git a/torchgeo/datamodules/glacier_calving_front.py b/torchgeo/datamodules/glacier_calving_front.py
@@ -53,15 +53,3 @@ def __init__(
         )
 
         self.size = size
-
-    def setup(self, stage: str) -> None:
-        """Set up datasets.
-
-        Args:
-            stage: Either 'fit', 'validate', 'test', or 'predict'.
-        """
-        if stage in ['fit', 'validate']:
-            self.train_dataset = GlacierCalvingFront(split='train', **self.kwargs)
-            self.val_dataset = GlacierCalvingFront(split='val', **self.kwargs)
-        if stage in ['test']:
-            self.test_dataset = GlacierCalvingFront(split='test', **self.kwargs)
diff --git a/torchgeo/datasets/glacier_calving_front.py b/torchgeo/datasets/glacier_calving_front.py
@@ -62,7 +62,7 @@ class GlacierCalvingFront(NonGeoDataset):
 
     mask_dirs = ('fronts', 'zones')
 
-    url = 'https://huggingface.co/datasets/torchgeo/glacier_calving_front/resolve/main/glacier_calving_data.zip'
+    url = 'https://huggingface.co/datasets/torchgeo/glacier_calving_front/resolve/fbf246c432a909ef96d96811b1f1b8a1e8c06542/glacier_calving_data.zip'
 
     md5 = '56e39e33f88a9842f48c513083e3c50a'