diff --git a/setup.cfg b/setup.cfg index f13bd0880339c4170c34f703bb97a5e5fdd56f3e..e5d685a532f2f0995a360dd19cd99de521339bba 100644 --- a/setup.cfg +++ b/setup.cfg @@ -36,7 +36,7 @@ install_requires = sympy>=0.5 # lower sympy version will have problems with a bug in floor when using interpolation jinja2 tqdm - p_tqdm # for parallel compilation in tensorflow_jit + p_tqdm>=1.3 # for parallel compilation in tensorflow_jit stringcase # for converting to camelcase for tensorflow # The usage of test_requires is discouraged, see `Dependency Management` docs test_requires = diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py index d223a505e65cfa58a60801e62cbad1377edd93c7..ebceb1889b94d55738c3549740933baf719fc317 100644 --- a/src/pystencils_autodiff/tensorflow_jit.py +++ b/src/pystencils_autodiff/tensorflow_jit.py @@ -198,7 +198,6 @@ def compile_sources_and_load(host_sources, object_files = [] sources = host_sources + cuda_sources - print('Compiling Tensorflow module...') def compile(source): is_cuda = source in cuda_sources @@ -221,7 +220,7 @@ def compile_sources_and_load(host_sources, return object_file # p_tqdm is just a parallel tqdm - object_files = p_tqdm.p_umap(compile, sources) + object_files = p_tqdm.p_umap(compile, sources, desc='Compiling Tensorflow module') print('Linking Tensorflow module...') module_file = link(object_files,