Skip to content

Commit

Permalink
Change how fit_dir is determined
Browse files Browse the repository at this point in the history
  • Loading branch information
OmegaLambda1998 committed Jan 30, 2024
1 parent 2a9094a commit a5d43d1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pippin/classifiers/supernnova.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,7 +270,7 @@ def classify(self, training):
light_curve_dir = sim_dep.output["photometry_dirs"][self.index]
self.raw_dir = light_curve_dir
fit = self.get_fit_dependency()
fit_dir = f"" if fit is None else f"--fits_dir {fit['fitres_dirs'][self.index]}"
fit_dir = f"" if ((fit is None) or (len(fit) == 0)) else f"--fits_dir {fit[self.index]['fitres_dirs']}"
cyclic = "--cyclic" if self.variant in ["vanilla", "variational"] and self.cyclic else ""
batch_size = f"--batch_size {self.batch_size}"
num_layers = f"--num_layers {self.num_layers}"
Expand Down

0 comments on commit a5d43d1

Please sign in to comment.