Fix deprecated tarfile call

MolSSI-BSE · Nov 10, 2023 · 0a59c5d · 0a59c5d
1 parent 6d8ca0d
commit 0a59c5d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/basis_set_exchange/tests/test_bundle.py b/basis_set_exchange/tests/test_bundle.py
@@ -33,6 +33,7 @@
 """
 
 import os
+import sys
 import zipfile
 import tarfile
 import pytest
@@ -50,7 +51,10 @@ def _extract_all(filepath, extract_dir):
             zf.extractall(extract_dir)
     elif filepath.endswith('.tar.bz2'):
         with tarfile.open(filepath, 'r:bz2') as tf:
-            tf.extractall(extract_dir)
+            if sys.version_info >= (3, 11):
+                tf.extractall(extract_dir, filter='fully_trusted')
+            else:
+                tf.extractall(extract_dir)
     else:
         raise RuntimeError("Unexpected file extension")