Merge pull request #377 from ahaidu-nv/ahaidu/boto3_fix

added boto3 import to s3 usage in utils.py
NVlabs · Jul 31, 2024 · cc88c0f · cc88c0f
2 parents 7544636 + c69d896
commit cc88c0f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/common/utils.py b/common/utils.py
@@ -191,6 +191,7 @@ def __init__(
         self.use_s3 = use_s3
 
         if self.use_s3:
+            import boto3
             self.session = boto3.Session()
             self.s3 = self.session.resource(
                 service_name="s3", endpoint_url=endpoint_url