From aca33af425424ef1780979405c6dc0d277577f14 Mon Sep 17 00:00:00 2001
From: Peter Luenenschloss <peter.luenenschloss@ufz.de>
Date: Fri, 10 Jul 2020 12:13:50 +0200
Subject: [PATCH] small bfx in data_modelling module
---
saqc/funcs/data_modelling.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/saqc/funcs/data_modelling.py b/saqc/funcs/data_modelling.py
index 6c3d1ade0..a58934574 100644
--- a/saqc/funcs/data_modelling.py
+++ b/saqc/funcs/data_modelling.py
@@ -153,7 +153,7 @@ def modelling_polyFit(data, field, flagger, winsz, polydeg, numba='auto', eval_f
to_flag = pd.Series(codes[num_cats.astype(int)], index=num_cats.index)
to_flag = to_flag.align(nan_samples)[0]
to_flag[nan_samples.index] = flags[nan_samples.index]
- flagger = flagger.setFlags(field, flags=to_flag, **kwargs)
+ flagger = flagger.setFlags(field, to_flag.values, **kwargs)
return data, flagger
@@ -228,6 +228,6 @@ def modelling_rollingMean(data, field, flagger, winsz, eval_flags=True, min_peri
to_flag = pd.Series(codes[num_cats.astype(int)], index=num_cats.index)
to_flag = to_flag.align(nan_samples)[0]
to_flag[nan_samples.index] = flags[nan_samples.index]
- flagger = flagger.setFlags(field, flags=to_flag, **kwargs)
+ flagger = flagger.setFlags(field, to_flag.values, **kwargs)
return data, flagger
--
GitLab