diff --git a/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py b/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py index 0b1bc97e5..184884bbe 100644 --- a/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py +++ b/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py @@ -23,8 +23,8 @@ def prv_forced_photometry_mapper() -> dict: mapping.update( { - "e_ra": Mapper(lambda x: 0), - "e_dec": Mapper(lambda x: 0), + "e_ra": Mapper(lambda: 0), + "e_dec": Mapper(lambda: 0), "isdiffpos": Mapper( lambda x: 1 if x >= 0 else -1, origin="forcediffimflux" ), diff --git a/sorting_hat_step/tests/unittest/test_wizard.py b/sorting_hat_step/tests/unittest/test_wizard.py index 829714c4c..03762548d 100644 --- a/sorting_hat_step/tests/unittest/test_wizard.py +++ b/sorting_hat_step/tests/unittest/test_wizard.py @@ -150,10 +150,10 @@ def test_decode(self): def test_write_object(self, insert_mock): alerts = pd.DataFrame( [ - {"oid": 10, "aid": 0, "sid": "ZTF", "extra": "extra1"}, - {"oid": 20, "aid": 1, "sid": "ZTF", "extra": "extra2"}, - {"oid": 30, "aid": 0, "sid": "ZTF", "extra": "extra3"}, - {"oid": 40, "aid": 2, "sid": "ZTF", "extra": "extra4"}, + {"oid": 10, "aid": 0, "sid": "ZTF", "extra_fields": {}}, + {"oid": 20, "aid": 1, "sid": "ZTF", "extra_fields": {}}, + {"oid": 30, "aid": 0, "sid": "ZTF", "extra_fields": {}}, + {"oid": 40, "aid": 2, "sid": "ZTF", "extra_fields": {}}, ] )