ant_fitting/fitting_all.py at master · AlexReedy/ant_fitting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from fittingLib import *
import timeit
fit = FittingLibrary(pause=2)

start = timeit.default_timer()
counter = 0
for data_set in fit.data_sets:
    ind_start = timeit.default_timer()

    fit.import_data(file=data_set)

    fit.plot_mag(save=True, show=True)
    fit.plot_flux(save=True, show=True)

    fit.sigma_clipping()
    fit.plot_sigma_clip(save=True, show=True)

    fit.get_average()
    fit.plot_sigma_clip_avg(save=True, show=True)
    fit.get_fit_parameters()

    ind_end = timeit.default_timer() - ind_start

    print(f'{data_set} [{counter} / {len(fit.data_sets)}] Completed:\n'
          f'Runtime: {int(ind_end)} sec\n'
          f'Moving to Next Data Set!')

    counter += 1

runtime = timeit.default_timer() - start

print(f'Total Runtime: {int(runtime)} sec')