Merge pull request #435 from utf/fix-aws-import

Fix aws module import
materialsproject · May 12, 2021 · 3d00e7f · 3d00e7f
2 parents 8472d59 + 6952076
commit 3d00e7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/maggma/stores/aws.py b/src/maggma/stores/aws.py
@@ -12,7 +12,6 @@
 from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
 
 import msgpack  # type: ignore
-from botocore.exceptions import ClientError
 from monty.dev import deprecated
 from monty.msgpack import default as monty_default
 
@@ -22,8 +21,9 @@
 try:
     import boto3
     import botocore
+    from botocore.exceptions import ClientError
     from boto3.session import Session
-except ImportError:
+except (ImportError, ModuleNotFoundError):
     boto3 = None  # type: ignore