fix init

janelia-cellmap · Mar 18, 2024 · d0989c3 · d0989c3
1 parent d85f13a
commit d0989c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cellmap_models/pytorch/cosem/__init__.py b/src/cellmap_models/pytorch/cosem/__init__.py
@@ -1,5 +1,5 @@
 from .load_model import load_model
-from .load_checkpoint import download_checkpoint
+from .download_checkpoint import download_checkpoint
 
 models_dict = {
     "setup04/1820500": "https://janelia-cosem-networks.s3.amazonaws.com/v0003.2-pytorch/cosem_models/cosem_models/setup04/1820500",