From 2a26d9c2db3644328a17ee998c44db0927d84667 Mon Sep 17 00:00:00 2001 From: Eddie Schlafly Date: Wed, 12 Jul 2023 11:22:28 -0400 Subject: [PATCH] Put context around model opening. --- src/stpipe/step.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/stpipe/step.py b/src/stpipe/step.py index 59fb9712..b98df599 100644 --- a/src/stpipe/step.py +++ b/src/stpipe/step.py @@ -825,12 +825,12 @@ def get_config_from_reference(cls, dataset, disable=None, crds_observatory=None) # If the dataset is not an operable instance of AbstractDataModel, # log as such and return an empty config object try: - model = cls._datamodels_open(dataset) - if isinstance(model, Sequence): - # Pull out first model in ModelContainer - model = model[0] - crds_parameters = model.get_crds_parameters() - crds_observatory = model.crds_observatory + with cls._datamodels_open(dataset) as model: + if isinstance(model, Sequence): + # Pull out first model in ModelContainer + model = model[0] + crds_parameters = model.get_crds_parameters() + crds_observatory = model.crds_observatory except (OSError, TypeError, ValueError): logger.warning("Input dataset is not an instance of AbstractDataModel.") disable = True