extend bug fix

arviz-devs · Jun 7, 2021 · 77f2a4c · 77f2a4c
1 parent 4d9caca
commit 77f2a4c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/arviz/data/inference_data.py b/arviz/data/inference_data.py
@@ -1394,10 +1394,11 @@ def extend(self, other, join="left"):
                 )
             dataset = getattr(other, group)
             setattr(self, group, dataset)
-            if group.startswith(WARMUP_TAG):
-                self._groups_warmup.append(group)
-            else:
-                self._groups.append(group)
+            if not hasattr(self, group):
+                if group.startswith(WARMUP_TAG):
+                    self._groups_warmup.append(group)
+                else:
+                    self._groups.append(group)
 
     set_index = _extend_xr_method(xr.Dataset.set_index, see_also="reset_index")
     get_index = _extend_xr_method(xr.Dataset.get_index)