added boto3 import

NVlabs · Jul 31, 2024 · c69d896 · c69d896
1 parent 7544636
commit c69d896
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