From 1c8d1f250463bb87a7994751dcd4844bd003c22c Mon Sep 17 00:00:00 2001 From: Emille Ishida Date: Fri, 1 Dec 2023 18:52:20 -0300 Subject: [PATCH] make import count inside test --- fink_science/random_forest_snia/processor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fink_science/random_forest_snia/processor.py b/fink_science/random_forest_snia/processor.py index c1ed56e2..c5ab23b7 100644 --- a/fink_science/random_forest_snia/processor.py +++ b/fink_science/random_forest_snia/processor.py @@ -12,7 +12,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from pyspark.sql.functions import pandas_udf, PandasUDFType, count +from pyspark.sql.functions import pandas_udf, PandasUDFType from pyspark.sql.types import DoubleType, StringType import pandas as pd @@ -620,7 +620,7 @@ def rfscore_rainbow_elasticc( # Perform the fit + classification (default model) >>> args = [F.col(i) for i in what_prefix] - >>> args += [count(F.col(what_prefix[0]))] + >>> args += [F.count(F.col(what_prefix[0]))] >>> args += [F.col('diaSource.snr')] >>> args += [F.col('diaObject.hostgal_snsep')] >>> args += [F.col('diaObject.hostgal_zphot')]