diff --git a/analysis/analyzers/render/__init__.py b/analysis/analyzers/render/__init__.py index aeb6a20..24c4d2d 100644 --- a/analysis/analyzers/render/__init__.py +++ b/analysis/analyzers/render/__init__.py @@ -5,7 +5,7 @@ from .. import Result class Render: result_types = [] - def render(self, results: List[Result], name=None): + def render(self, results: List[Result], name=None) -> [str]: raise NotImplementedError() def filter(self, results: List[Result]): diff --git a/analysis/analyzers/render/biogames.py b/analysis/analyzers/render/biogames.py index 16e8b63..53c12c4 100644 --- a/analysis/analyzers/render/biogames.py +++ b/analysis/analyzers/render/biogames.py @@ -135,16 +135,16 @@ class BoardDurationHistRender(Render): class BoardDurationBoxRender(Render): result_types = [BoardDurationAnalyzer] - def render(self, results: List[Result], name=None): + def render(self, results: List[Result], name=None) -> [str]: data = defaultdict(list) for result in self.filter(results): - get = result.get() - for board in get: + for board in result.get(): duration = board['active'] if 'active' in board else 0 data[board['id']].append(duration) data_tuples = [(key, data[key]) for key in sorted(data)] data_tuples = sorted(data_tuples, key=lambda x: sum(x[1])) - plot(data_tuples) + plot(data_tuples, name=name) + return [name] class ActivityMapperRender(Render):