cache

runner/inference.py

@@ -206,13 +206,12 @@ def download_infercence_cache(configs: Any, model_version: str = "v0.2.0") -> No
 
     data_cache_dir = os.path.join(code_directory, "release_data/ccd_cache")
     os.makedirs(data_cache_dir, exist_ok=True)
-    for cache_name, fname in [
-        ("ccd_components_file", "components.v20240608.cif"),
-    ]:
-        if not opexists(cache_path := os.path.abspath(opjoin(data_cache_dir, fname))):
-            tos_url = URL[cache_name]
-            logger.info(f"Downloading data cache from\n {tos_url}...")
-            urllib.request.urlretrieve(tos_url, cache_path)
+
+    if not os.path.exists('./release_data/ccd_cache/components.v20240608.cif'):
+        file_id = '1e8wxpuEB-0xL_3dlMfZCFo6cL5oSHSUK'
+        download_url = f'https://drive.google.com/uc?id={file_id}'
+        output_file = './release_data/ccd_cache/components.v20240608.cif'
+        gdown.download(download_url, output_file, quiet=False)
 
     if not os.path.exists('./release_data/ccd_cache/components.v20240608.cif.rdkit_mol.pkl'):
         file_id = '1R9d678aBfQwTd0Rh15doRmW-fETNdeWf'