diff --git a/pytorch3d/implicitron/dataset/types.py b/pytorch3d/implicitron/dataset/types.py index 497b91eee..b8227567a 100644 --- a/pytorch3d/implicitron/dataset/types.py +++ b/pytorch3d/implicitron/dataset/types.py @@ -226,7 +226,7 @@ def _dataclass_list_from_dict_list(dlist, typeannot): keys = np.split(list(all_keys_res), indices[:-1]) vals = np.split(list(all_vals_res), indices[:-1]) - return [cls(zip(*k, v)) for k, v in zip(keys, vals)] + return [cls(zip(k, v)) for k, v in zip(keys, vals)] elif not dataclasses.is_dataclass(typeannot): return dlist diff --git a/tests/implicitron/test_types.py b/tests/implicitron/test_types.py index 56352b93b..aff749be5 100644 --- a/tests/implicitron/test_types.py +++ b/tests/implicitron/test_types.py @@ -73,8 +73,8 @@ def test_parsing(self): ) # dict - parsed = types._dataclass_from_dict({"k": dct}, Dict[str, FrameAnnotation]) - self.assertEqual(parsed, {"k": self.entry}) + parsed = types._dataclass_from_dict({"key": dct}, Dict[str, FrameAnnotation]) + self.assertEqual(parsed, {"key": self.entry}) def test_parsing_vectorized(self): dct = dataclasses.asdict(self.entry) @@ -83,10 +83,10 @@ def test_parsing_vectorized(self): self._compare_with_scalar(_NT(dct), _NT) self._compare_with_scalar((dct,), Tuple[FrameAnnotation]) self._compare_with_scalar([dct], List[FrameAnnotation]) - self._compare_with_scalar({"k": dct}, Dict[str, FrameAnnotation]) + self._compare_with_scalar({"key": dct}, Dict[str, FrameAnnotation]) dct2 = dct.copy() - dct2["meta"] = {"d": 76} + dct2["meta"] = {"aux": 76} self._compare_with_scalar(dct2, FrameAnnotation) def _compare_with_scalar(self, obj, typeannot, repeat=3):