Coverage for jstark/features/sum_feature.py: 100%

1"""Sum abstract base class""" 

2from abc import ABCMeta 

3from typing import Callable 

4 

5import pyspark.sql.functions as f 

6from pyspark.sql import Column 

7 

8 

9from .feature import BaseFeature 

10 

11 

12class Sum(BaseFeature, metaclass=ABCMeta): 

13 def aggregator(self) -> Callable[[Column], Column]: 

14 return self.sum_aggregator 

15 

16 def default_value(self) -> Column: 

17 return f.lit(0.0)