diff --git a/case_studies/redshift/redshift_from_img/encoder/encoder.py b/case_studies/redshift/redshift_from_img/encoder/encoder.py index f21c8c5c1..cd18c9f13 100644 --- a/case_studies/redshift/redshift_from_img/encoder/encoder.py +++ b/case_studies/redshift/redshift_from_img/encoder/encoder.py @@ -13,7 +13,7 @@ def __init__( ): super().__init__(*args, **kwargs) self.discrete_metrics = discrete_metrics - + def update_metrics(self, batch, batch_idx): target_cat = TileCatalog(batch["tile_catalog"]).get_brightest_sources_per_tile()