@@ -196,25 +196,30 @@ def main():
196196 values = spl [1 ].split ("," )
197197 for j in range (len (values )):
198198 values [j ] = values [j ].strip ()
199- if values [2 ][0 ] == 'p' :
200- if mode != 1 :
201- raise IOError ("Random mode must be used when passing predefined priors" )
202- if values [0 ] not in prior_files :
203- prior_files .append (values [0 ])
204- if values [1 ] == 'npy' :
205- fprior = np .load (values [0 ])
206- prior_index = np .random .choice (fprior .shape [0 ], size = randsize , replace = False )
207- prior_indicies .append (prior_index )
208- samp = fprior [prior_index ]
209- prior_samples .append (samp )
210- elif values [1 ] == 'txt' or values [1 ] == 'dat' :
211- fprior = ascii .read (values [0 ])
212- prior_index = np .random .choice (len (fprior ), size = randsize , replace = False )
213- prior_indicies .append (prior_index )
214- samp = fprior [prior_index ]
215- prior_samples .append (samp )
216- elif values [1 ] != 'npy' and values [1 ] != 'txt' and values [1 ] != 'dat' :
217- raise IOError ("File type incompatible for predefined prior mode. Acceptable file types: npy, ascii formatted txt, ascii formatted dat" )
199+ try :
200+ if values [2 ][0 ] == 'p' :
201+ if mode != 1 :
202+ raise IOError ("Random mode must be used when passing predefined priors" )
203+ if values [0 ] not in prior_files :
204+ prior_files .append (values [0 ])
205+ if values [1 ] == 'npy' :
206+ fprior = np .load (values [0 ])
207+ prior_index = np .random .choice (fprior .shape [0 ], size = randsize , replace = False )
208+ prior_indicies .append (prior_index )
209+ samp = fprior [prior_index ]
210+ prior_samples .append (samp )
211+ elif values [1 ] == 'txt' or values [1 ] == 'dat' :
212+ fprior = ascii .read (values [0 ])
213+ prior_index = np .random .choice (len (fprior ), size = randsize , replace = False )
214+ prior_indicies .append (prior_index )
215+ samp = fprior [prior_index ]
216+ prior_samples .append (samp )
217+ elif values [1 ] != 'npy' and values [1 ] != 'txt' and values [1 ] != 'dat' :
218+ raise IOError ("File type incompatible for predefined prior mode. Acceptable file types: npy, ascii formatted txt, ascii formatted dat" )
219+ except :
220+ print ('ERROR: Incorrect number of arguments.' )
221+ print (inputf + ', line ' + i + ': ' + lines [i ])
222+ exit ()
218223 # End pass 1a through input file -------------------------------
219224 # End Megans Addition ------------------------------------------
220225
0 commit comments