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

1"""Sum abstract base class""" 

2 

3from abc import ABCMeta 

4from typing import Callable 

5 

6import pyspark.sql.functions as f 

7from pyspark.sql import Column 

8 

9 

10from .feature import BaseFeature 

11 

12 

13class Sum(BaseFeature, metaclass=ABCMeta): 

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

15 return self.sum_aggregator 

16 

17 def default_value(self) -> Column: 

18 return f.lit(0.0)