Make use of p_tqdm==1.3

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 =

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,