Merge pull request #25 from haddocking/logging

Fixing the logging issue reported by Richard
haddocking · Dec 8, 2023 · f3b46a2 · f3b46a2
2 parents deb981d + cd9fecb
commit f3b46a2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/disvis/disvis.py b/disvis/disvis.py
@@ -388,10 +388,11 @@ def _print_progress(n, total, time0):
         m = n + 1
         pdone = m/total
         t = _time() - time0
-        _stdout.write('\r{:d}/{:d} ({:.2%}, ETA: {:d}s)    '\
-                .format(m, total, pdone, 
-                        int(t/pdone - t)))
-        _stdout.flush()
+        if _stdout.isatty():
+            _stdout.write('\r{:d}/{:d} ({:.2%}, ETA: {:d}s\n)    '\
+                    .format(m, total, pdone, 
+                            int(t/pdone - t)))
+            _stdout.flush()
 
     def _gpu_init(self):
 

diff --git a/disvis/main.py b/disvis/main.py
@@ -339,8 +339,7 @@ def run_disvis(queue, receptor, ligand, rotmat, weights, distance_restraints,
 
 
 def write(line):
-    if stdout.isatty():
-        print(line)
+    print(line)
     logging.info(line)