Skip to content
Snippets Groups Projects
Commit f1e685c7 authored by David Schäfer's avatar David Schäfer
Browse files

pass the test function names to the flagger

parent 82ec3024
No related branches found
No related tags found
No related merge requests found
......@@ -130,14 +130,14 @@ def runner(metafname, flagger, data, flags=None, nodata=np.nan):
if Params.FLAGPERIOD in flag_params:
ffchunk = assignTypeSafe(
ffchunk, varname,
flagPeriod(flagger, ffchunk[varname], mask, **flag_params))
flagPeriod(flagger, ffchunk[varname], mask, func_name=func_name, **flag_params))
# flag a certain amount of values after condition is met
if Params.FLAGVALUES in flag_params:
ffchunk = assignTypeSafe(
ffchunk,
varname,
flagNext(flagger, ffchunk[varname], mask, **flag_params))
flagNext(flagger, ffchunk[varname], mask, func_name=func_name, **flag_params))
if flag_params.get(Params.PLOT, False):
plotvars.append(varname)
......
......@@ -71,9 +71,6 @@ class DmpFlagger(BaseFlagger):
flagcol = flags.loc[:, FlagFields.FLAG].squeeze()
return super().isFlagged(flagcol, flag, comparator)
def _getCaller(self):
return inspect.stack()[1].function
def _reduceColumns(self, flags):
if set(flags.columns) == set(self.flag_fields):
pass
......
#! /usr/bin/env python
# -*- coding: utf-8 -*-
import logging
from functools import partial
import numpy as np
import pandas as pd
import logging
from scipy.signal import savgol_filter
......@@ -26,8 +28,8 @@ FUNC_MAP = {}
def register(name):
def outer(func):
func = partial(func, func_name=name)
FUNC_MAP[name] = func
func.__name__ = name
def inner(*args, **kwargs):
return func(*args, **kwargs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment