Skip to content

Commit

Permalink
fix TOI notebook hardcoded path
Browse files Browse the repository at this point in the history
  • Loading branch information
avivajpeyi committed Aug 7, 2023
1 parent d4e1166 commit eed2dfb
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions src/tess_atlas/notebook_controllers/templates/toi_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,11 +100,10 @@
plot_raw_lightcurve,
)

# TOI_NUMBER = {{{TOINUMBER}}}
TOI_NUMBER = 103
TOI_NUMBER = {{{TOINUMBER}}}
logger = get_notebook_logger(outdir=f"toi_{TOI_NUMBER}_files")

# + tags=["exe", "hide-cell"]
# + tags=["exe", "remove-cell"]
import theano

from tess_atlas.utils import tabulate_global_environ_vars
Expand All @@ -122,7 +121,7 @@
#
# - The `TIC Entry` object holds one or more `Planet Candidate`s (each candidate associated with one TOI id number) and a `Lightcurve Data` for associated with the candidates. Note that the `Lightcurve Data` object is initially the same fopr each candidate but may be masked according to the candidate transit's period.
#
# - The `Planet Candidate` holds informaiton on the TOI data collected by [SPOC] (eg transit period, etc)
# - The `Planet Candidate` holds information on the TOI data collected by [SPOC] (eg transit period, etc)
#
# - The `Lightcurve Data` holds the lightcurve time and flux data for the planet candidates.
#
Expand Down Expand Up @@ -301,7 +300,7 @@
def run_inference(model) -> InferenceData:
np.random.seed(TOI_NUMBER)
with model:
sampling_kwargs = dict(tune=200, draws=200, chains=1, cores=1)
sampling_kwargs = dict(tune=2000, draws=2000, chains=2, cores=2)
logger.info(f"Run sampler with kwargs: {sampling_kwargs}")
inference_data = pmx.sample(
**sampling_kwargs, start=init_params, return_inferencedata=True
Expand Down

0 comments on commit eed2dfb

Please sign in to comment.