Skip to content

Commit

Permalink
fix test.
Browse files Browse the repository at this point in the history
I manually checked the first 20 objects and although they are in a different order, they agree until the 5th decimal number.
  • Loading branch information
emilleishida authored Dec 6, 2023
1 parent c92b46a commit 5e8f5d6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fink_science/random_forest_snia/processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -474,9 +474,9 @@ def rfscore_rainbow_elasticc(
>>> args += [F.col('diaObject.hostgal_snsep')]
>>> args += [F.col('diaObject.hostgal_zphot')]
>>> df = df.withColumn('pIa', rfscore_rainbow_elasticc(*args))
>>> df.select('pIa').show()
>>> df.filter(df['pIa'] > 0.5).count()
80
79
"""
# dt is a column of floats
dt = midPointTai.apply(lambda x: np.max(x) - np.min(x))
Expand Down

0 comments on commit 5e8f5d6

Please sign in to comment.