fix synapse data download interface (#426)

theislab · Nov 29, 2021 · 963d043 · 963d043
1 parent c628034
commit 963d043
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sfaira/data/dataloaders/base/dataset.py b/sfaira/data/dataloaders/base/dataset.py
@@ -350,8 +350,8 @@ def _download_synapse(self, synapse_entity, fn, **kwargs):
         print(f"Downloading from synapse: {fn}")
         syn = synapseclient.Synapse()
         syn.login(kwargs['synapse_user'], kwargs['synapse_pw'])
-        dataset = syn.get(entity=synapse_entity)
-        shutil.move(dataset.data_dir_base, os.path.join(self.data_dir, fn))
+        syn.get(entity=synapse_entity, downloadLocation=os.path.join(self.data_dir, fn))
+        syn.logout()
 
     @property
     def cache_fn(self):