Added missing imports

Coda-Research-Group · Mar 18, 2024 · c225917 · c225917
1 parent 7d0ddb6
commit c225917
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/search/tests/test_insert.py b/search/tests/test_insert.py
@@ -8,6 +8,10 @@
 import numpy as np
 import h5py
 import os
+from pathlib import Path
+from urllib.request import urlretrieve
+import logging
+
 
 def get_data(data_part, **config):
     return np.array(
@@ -25,7 +29,7 @@ def get_data(data_part, **config):
 def download(src, dst):
     if not os.path.exists(dst):
         os.makedirs(Path(dst).parent, exist_ok=True)
-        LOG.info("downloading %s -> %s..." % (src, dst))
+        logging.info("downloading %s -> %s..." % (src, dst))
         urlretrieve(src, dst)