Fixing lint problems.

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

Hide file tree

Showing 2 changed files with 8 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, filter='data'):
+def extract_archive(file, target_dir, overwrite=True, filter="data"):
     """Extract archive file.
 
     Parameters
@@ -282,7 +282,9 @@ 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, filter=filter)
+                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, filter='data'):
+def extract_archive(file, target_dir, overwrite=True, filter="data"):
     """Extract archive file.
 
     Parameters
@@ -351,7 +351,9 @@ 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, filter=filter)
+                tar.extractall(
+                    path, members, numeric_owner=numeric_owner, filter=filter
+                )
 
             safe_extract(archive, path=target_dir)
     elif file.endswith(".gz"):