[Fix] ConcatDataset raises error when metainfo is np.array (#1407)

open-mmlab · Oct 31, 2023 · 2a563f4 · 2a563f4
1 parent e0cf958
commit 2a563f4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmengine/dataset/dataset_wrapper.py b/mmengine/dataset/dataset_wrapper.py
@@ -84,7 +84,8 @@ def __init__(self,
                 if (isinstance(self._metainfo[key], np.ndarray)
                         and not np.array_equal(self._metainfo[key],
                                                dataset.metainfo[key])
-                        or self._metainfo[key] != dataset.metainfo[key]):
+                        or (not isinstance(self._metainfo[key], np.ndarray)
+                            and self._metainfo[key] != dataset.metainfo[key])):
                     raise ValueError(
                         f'The meta information of the {i}-th dataset does not '
                         'match meta information of the first dataset')