Skip to content

Commit

Permalink
Update syntax for AL Ia for elasticc (requires fink-science>=5.50)
Browse files Browse the repository at this point in the history
  • Loading branch information
JulienPeloton committed Dec 7, 2023
1 parent d552757 commit c22bcae
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions fink_broker/science.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@
from fink_science.ad_features.processor import extract_features_ad
from fink_science.anomaly_detection.processor import anomaly_score

from fink_science.random_forest_snia.processor import rfscore_sigmoid_elasticc
from fink_science.random_forest_snia.processor import rfscore_rainbow_elasticc
from fink_science.snn.processor import snn_ia_elasticc, snn_broad_elasticc
from fink_science.cats.processor import predict_nn
from fink_science.agn.processor import agn_elasticc
Expand Down Expand Up @@ -450,15 +450,11 @@ def apply_science_modules_elasticc(df: DataFrame) -> DataFrame:

_LOG.info("New processor: EarlySN")
args = ['cmidPointTai', 'cfilterName', 'cpsFlux', 'cpsFluxErr']

# fake cdsxmatch and nobs
args += [F.col('diaObject.ra'), F.col('diaObject.decl')]
args += [F.col('diaObject.hostgal_ra'), F.col('diaObject.hostgal_dec')]
args += [F.col('diaSource.snr')]
args += [F.col('diaObject.hostgal_snsep')]
args += [F.col('diaObject.hostgal_zphot')]
args += [F.col('diaObject.hostgal_zphot_err')]

df = df.withColumn('rf_snia_vs_nonia', rfscore_sigmoid_elasticc(*args))
df = df.withColumn('rf_snia_vs_nonia', rfscore_rainbow_elasticc(*args))

# Apply level one processor: superNNova
_LOG.info("New processor: supernnova - Ia")
Expand Down

0 comments on commit c22bcae

Please sign in to comment.