Add SSL cert installation to resultset

rapidsai · Dec 10, 2024 · f3d896a · f3d896a
1 parent bfcf422
commit f3d896a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/cugraph/cugraph/testing/resultset.py b/python/cugraph/cugraph/testing/resultset.py
@@ -13,16 +13,14 @@
 
 import warnings
 import tarfile
-
 import urllib.request
 
 import cudf
 from cugraph.datasets.dataset import (
     DefaultDownloadDir,
     default_download_dir,
 )
-
-# results_dir_path = utils.RAPIDS_DATASET_ROOT_DIR_PATH / "tests" / "resultsets"
+from cugraph.utilities import install_ssl_cert
 
 
 class Resultset:
@@ -107,6 +105,7 @@ def load_resultset(resultset_name, resultset_download_url):
         if not curr_resultset_download_dir.exists():
             curr_resultset_download_dir.mkdir(parents=True, exist_ok=True)
         if not compressed_file_path.exists():
+            install_ssl_cert()
             urllib.request.urlretrieve(resultset_download_url, compressed_file_path)
         tar = tarfile.open(str(compressed_file_path), "r:gz")
         # TODO: pass filter="fully_trusted" when minimum supported Python version >=3.12