From 06cfa92211db69d211eb449514977419daab4d9e Mon Sep 17 00:00:00 2001 From: Ari Hartikainen Date: Sun, 16 Jan 2022 11:44:06 +0200 Subject: [PATCH] Fix long lines --- arviz/data/inference_data.py | 28 ++++++++++++++++++++++------ 1 file changed, 22 insertions(+), 6 deletions(-) diff --git a/arviz/data/inference_data.py b/arviz/data/inference_data.py index 348d73f318..3c609d06a2 100644 --- a/arviz/data/inference_data.py +++ b/arviz/data/inference_data.py @@ -1358,9 +1358,15 @@ def add_groups(self, group_dict=None, coords=None, dims=None, **kwargs): if dataset: setattr(self, group, dataset) if group.startswith(WARMUP_TAG): - supported_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup] + supported_order = [ + key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup + ] if (supported_order == self._groups_warmup) and (group in SUPPORTED_GROUPS_ALL): - group_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup + [group]] + group_order = [ + key + for key in SUPPORTED_GROUPS_ALL + if key in self._groups_warmup + [group] + ] group_idx = group_order.index(group) self._groups_warmup.insert(group_idx, group) else: @@ -1368,7 +1374,9 @@ def add_groups(self, group_dict=None, coords=None, dims=None, **kwargs): else: supported_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups] if (supported_order == self._groups) and (group in SUPPORTED_GROUPS_ALL): - group_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups + [group]] + group_order = [ + key for key in SUPPORTED_GROUPS_ALL if key in self._groups + [group] + ] group_idx = group_order.index(group) self._groups.insert(group_idx, group) else: @@ -1408,9 +1416,15 @@ def extend(self, other, join="left"): setattr(self, group, dataset) if group.startswith(WARMUP_TAG): if group not in self._groups_warmup: - supported_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup] + supported_order = [ + key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup + ] if (supported_order == self._groups_warmup) and (group in SUPPORTED_GROUPS_ALL): - group_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups_warmup + [group]] + group_order = [ + key + for key in SUPPORTED_GROUPS_ALL + if key in self._groups_warmup + [group] + ] group_idx = group_order.index(group) self._groups_warmup.insert(group_idx, group) else: @@ -1419,7 +1433,9 @@ def extend(self, other, join="left"): if group not in self._groups: supported_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups] if (supported_order == self._groups) and (group in SUPPORTED_GROUPS_ALL): - group_order = [key for key in SUPPORTED_GROUPS_ALL if key in self._groups + [group]] + group_order = [ + key for key in SUPPORTED_GROUPS_ALL if key in self._groups + [group] + ] group_idx = group_order.index(group) self._groups.insert(group_idx, group) else: