From 7ebdac90db16175868aa7a8eb9d9a06f1c492452 Mon Sep 17 00:00:00 2001
From: James Meakin <12661555+jmsmkn@users.noreply.github.com>
Date: Tue, 6 Feb 2024 16:00:09 +0100
Subject: [PATCH] Be liberal with the compression types

---
 sagemaker_shim/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sagemaker_shim/models.py b/sagemaker_shim/models.py
index bbf25c5..06db2df 100644
--- a/sagemaker_shim/models.py
+++ b/sagemaker_shim/models.py
@@ -94,7 +94,7 @@ def download_and_extract_tarball(*, s3_uri: str, dest: Path) -> None:
 
         f.seek(0)
 
-        with tarfile.open(fileobj=f, mode="r:gz") as tar:
+        with tarfile.open(fileobj=f, mode="r") as tar:
             tar.extractall(path=dest, filter="data")