From d01ed0fedf77de5f9bcbd349e9e64933ac290c6c Mon Sep 17 00:00:00 2001 From: Ricky Sexton Date: Thu, 21 Nov 2024 16:07:21 -0700 Subject: [PATCH] removed plot_protein fail if tm_dict.txt not found --- basicrta/util.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/basicrta/util.py b/basicrta/util.py index 9b2e33e..98d8232 100644 --- a/basicrta/util.py +++ b/basicrta/util.py @@ -401,9 +401,12 @@ def make_residue_plots(results, comps=None, show=False): def plot_protein(residues, t_slow, bars, prot=None, label_cutoff=3, ylim=None, major_tick=None, minor_tick=None, scale=1): - with open('tm_dict.txt', 'r') as f: - contents = f.read() - prots = ast.literal_eval(contents) + try: + with open('tm_dict.txt', 'r') as f: + contents = f.read() + prots = ast.literal_eval(contents) + except FileNotFoundError("tm_dict.txt not found"): + prot = None if not os.path.exists('figs'): os.mkdir('figs')