Fix checking HTTP status code for boto3 responses

allenai · Mar 13, 2024 · 0b6e28c · 0b6e28c
1 parent 0b835a8
commit 0b6e28c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/olmo/util.py b/olmo/util.py
@@ -550,7 +550,7 @@ def _s3_get_bytes_range(
                 .read()
             )
         except boto_exceptions.ClientError as e:
-            if int(e.response["Error"]["Code"]) == 404:
+            if e.response["ResponseMetadata"]["HTTPStatusCode"] == 404:
                 raise FileNotFoundError(f"s3://{bucket_name}/{key}") from e
             err = e
         except (boto_exceptions.HTTPClientError, boto_exceptions.ConnectionError) as e: