diff --git a/nmma/tests/injections.py b/nmma/tests/injections.py index c101446a..adb4dfd4 100644 --- a/nmma/tests/injections.py +++ b/nmma/tests/injections.py @@ -50,7 +50,10 @@ def create_injection_from_command_line(model_name): path to the injection file created by nmma_create_injection """ - prior_path = os.path.join("./priors/", model_name + ".prior") + if model_name == "nugent-hyper": + prior_path = os.path.join("./priors/", "sncosmo-generic" + ".prior") + else: + prior_path = os.path.join("./priors/", model_name + ".prior") assert os.path.exists(prior_path), "prior file does not exist" injection_name = os.path.join(test_directory, model_name + "_injection.json")