Resolve Warnings for Extraction Filters.

dmlc · Nov 7, 2024 · a13a6b3 · a13a6b3
1 parent 88f109f
commit a13a6b3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/dgl/data/utils.py b/python/dgl/data/utils.py
@@ -244,7 +244,7 @@ def check_sha1(filename, sha1_hash):
     return sha1.hexdigest() == sha1_hash
 
 
-def extract_archive(file, target_dir, overwrite=True):
+def extract_archive(file, target_dir, overwrite=True, filter='data'):
     """Extract archive file.
 
     Parameters
@@ -282,7 +282,7 @@ def safe_extract(
                     member_path = os.path.join(path, member.name)
                     if not is_within_directory(path, member_path):
                         raise Exception("Attempted Path Traversal in Tar File")
-                tar.extractall(path, members, numeric_owner=numeric_owner)
+                tar.extractall(path, members, numeric_owner=numeric_owner, filter=filter)
 
             safe_extract(archive, path=target_dir)
     elif file.endswith(".gz"):

diff --git a/python/dgl/graphbolt/internal_utils.py b/python/dgl/graphbolt/internal_utils.py
@@ -313,7 +313,7 @@ def check_sha1(filename, sha1_hash):
     return sha1.hexdigest() == sha1_hash
 
 
-def extract_archive(file, target_dir, overwrite=True):
+def extract_archive(file, target_dir, overwrite=True, filter='data'):
     """Extract archive file.
 
     Parameters
@@ -351,7 +351,7 @@ def safe_extract(
                     member_path = os.path.join(path, member.name)
                     if not is_within_directory(path, member_path):
                         raise Exception("Attempted Path Traversal in Tar File")
-                tar.extractall(path, members, numeric_owner=numeric_owner)
+                tar.extractall(path, members, numeric_owner=numeric_owner, filter=filter)
 
             safe_extract(archive, path=target_dir)
     elif file.endswith(".gz"):