printing

int-brain-lab · Jul 20, 2023 · a65f65b · a65f65b
1 parent 9a502e3
commit a65f65b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/needles2/probe_model.py b/needles2/probe_model.py
@@ -355,7 +355,9 @@ def crawl_up_from_tip(ins, covered_length):
         per0 = []
 
         for p, traj in enumerate(trajs):
-            print(f'Trajectory: {traj["id"]}, provenance: {traj["provenance"]}, PID: {traj["probe_insertion"]}')
+            keys_display = ["id", "provenance", "probe_insertion"]  # Keys to print, if exist
+            if len(set(keys_display).intersection(traj.keys())) == len(keys_display):
+                print(f'Trajectory: {traj["id"]}, provenance: {traj["provenance"]}, PID: {traj["probe_insertion"]}')
             if len(trajs) > 20 and self.verbose is True:
                 if p % 20 == 0:
                     print(p / len(trajs))