Skip to content

Commit

Permalink
absorption_function.py: update argument description
Browse files Browse the repository at this point in the history
  • Loading branch information
jannisteunissen committed Nov 22, 2023
1 parent c4c327a commit 58ed978
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions tools/absorption_function.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,13 +29,15 @@ def get_args():
p.add_argument('-n_modes', type=int, default=3,
help='Number of Helmholtz modes')
p.add_argument('-H2O_model', type=str, choices=['Naidis', 'Aints'],
default='Naidis', help='Type of H2O absorption model')
default='Naidis',
help='H2O absorption coefficients for '
'numerical absorption function')
p.add_argument('-guess_amplitudes', type=float,
help='Initial guess for amplitude of Helmholtz modes')
p.add_argument('-guess_lambdas', type=float,
help='Initial guess for lambdas of Helmholtz modes')
p.add_argument('-fit_what', type=str, default='numerical',
choices=['numerical', 'Zheleznyak', 'Aints'],
choices=['numerical', 'Zheleznyak-H2O', 'Aints'],
help='What type of data/function to fit')
p.add_argument('-fit_type', type=str, default='least_squares',
choices=['least_squares', 'relative', 'log'],
Expand Down Expand Up @@ -222,7 +224,7 @@ def integrand(x, r):

if args.fit_what == 'numerical':
f_to_fit = f_numerical
elif args.fit_what == 'Zheleznyak':
elif args.fit_what == 'Zheleznyak-H2O':
f_to_fit = naidis_moist_zheleznyak_f(r)
elif args.fit_what == 'Aints':
f_to_fit = aints_moist_zheleznyak_f(r)
Expand Down

0 comments on commit 58ed978

Please sign in to comment.