Commit cc4f17af authored by Sascha Herzinger's avatar Sascha Herzinger
Browse files

some test refactoring and a bugfix regarding meta data

parent 2ebbd81a
Pipeline #2272 failed with stage
in 6 minutes and 16 seconds
......@@ -108,9 +108,10 @@ class ETL(Task, metaclass=abc.ABCMeta):
value = redis.get(name='data:{}'.format(self.request.id))
assert value is not None
data_state = json.loads(value)
features = data_frame.get('feature')
if features:
features = features.unique().tolist()
if 'feature' in data_frame.columns:
features = data_frame['feature'].unique().tolist()
else:
features = []
data_state['loaded'] = True
data_state['meta']['features'] = features
redis.setex(name='data:{}'.format(self.request.id),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment