Updated weight download test

Author: C-Achard

napari_cellseg3d/_tests/test_weight_download.py

@@ -2,16 +2,10 @@
     WeightsDownloader,
     WEIGHTS_DIR,
 )
-from napari_cellseg3d.config import ModelInfo
-
 
 def test_weight_download():
-
-    info = ModelInfo()
-
     downloader = WeightsDownloader()
-
-    downloader.download_weights(info.name, info.get_model().get_weights_file())
-    result_path = WEIGHTS_DIR / str(info.get_model().get_weights_file())
+    downloader.download_weights("test", "test.pth")
+    result_path = WEIGHTS_DIR / "test.pth"
 
     assert result_path.is_file()

napari_cellseg3d/code_models/models/pretrained/pretrained_model_urls.json

@@ -2,5 +2,6 @@
   "TRAILMAP_MS": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/TRAILMAP_MS.tar.gz",
   "SegResNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/SegResNet.tar.gz",
   "VNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/VNet.tar.gz",
-  "SwinUNetR": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/Swin64.tar.gz"
+  "SwinUNetR": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/Swin64.tar.gz",
+  "test": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/test.tar.gz"
 }