diff --git a/barrOs.py b/barrOs.py index e27ab7df375a0e4827214695e959ce6ab581bbf4..e934064fb3ef3a5eacabca5f2a20f6dc19faa37d 100644 --- a/barrOs.py +++ b/barrOs.py @@ -52,9 +52,11 @@ if __name__ == '__main__': if len(pdbIDs) > 1: for param in data.columns: if 'mad' not in param and param != 'PDBs' and param != 'TMsegm': - barros.plot_parameter('TMsegm', param, data.loc[data.N_chains == 1], fit_line = True, saveto = '{}_{}_monomers.pdf'.format(outf.split('.')[-2], param)) - barros.plot_parameter('TMsegm', param, data.loc[data.N_chains > 1], fit_line = True, saveto = '{}_{}_multimers.pdf'.format(outf.split('.')[-2], param)) - + try: + barros.plot_parameter('TMsegm', param, data.loc[data.N_chains == 1], fit_line = True, saveto = '{}_{}_monomers.pdf'.format(outf.split('.')[-2], param)) + barros.plot_parameter('TMsegm', param, data.loc[data.N_chains > 1], fit_line = True, saveto = '{}_{}_multimers.pdf'.format(outf.split('.')[-2], param)) + except: + print('ERROR: Not enough values to plot for', param)