Fix long lines

arviz-devs · Jan 16, 2022 · 06cfa92 · 06cfa92
1 parent ec699c3
commit 06cfa92
Showing 1 changed file with 22 additions and 6 deletions.
diff --git a/arviz/data/inference_data.py b/arviz/data/inference_data.py
@@ -1358,17 +1358,25 @@ 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:
                         self._groups_warmup.append(group)
                 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: