From c925d86c044d312130ba780ec02d817870470460 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Matthias=20K=C3=BCmmmerer?= Date: Thu, 2 May 2024 21:19:09 +0200 Subject: [PATCH] Cleanup output when exporting HDF5 models MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Matthias Kümmmerer --- pysaliency/precomputed_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pysaliency/precomputed_models.py b/pysaliency/precomputed_models.py index 795e5b7..1c3ba08 100644 --- a/pysaliency/precomputed_models.py +++ b/pysaliency/precomputed_models.py @@ -1,6 +1,7 @@ from __future__ import absolute_import, division, print_function import glob +import logging import os.path import tarfile import warnings @@ -102,7 +103,7 @@ def export_model_to_hdf5(model, stimuli, filename, compression=9, overwrite=True with h5py.File(filename, mode=mode) as f: for k, s in enumerate(tqdm(stimuli)): if not overwrite and names[k] in f: - print("Skipping already existing entry", names[k]) + logging.debug(f"Skipping already existing entry {names[k]}") continue if isinstance(model, SaliencyMapModel): smap = model.saliency_map(s)