Update constants.py (microsoft#24)

Fixes microsoft#23.
hyoo · May 3, 2022 · 3591201 · 3591201
1 parent 478019c
commit 3591201
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metaseq/service/constants.py b/metaseq/service/constants.py
@@ -15,7 +15,7 @@
 
 try:
     from metaseq_internal.constants import LOCAL_SSD, MODEL_SHARED_FOLDER
-except ImportError:
+except ModuleNotFoundError:
     # MODEL_SHARED_FOLDER should point to a shared drive (e.g. NFS) where the
     # checkpoints from S3 are stored. As an example:
     # MODEL_SHARED_FOLDER = "/example/175B/reshard_no_os"