We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0bf0efa commit aa1bcddCopy full SHA for aa1bcdd
tests/test_meta_tensor.py
@@ -74,8 +74,8 @@ def check_meta(self, a: MetaTensor, b: MetaTensor) -> None:
74
aff_a = meta_a.get("affine", None)
75
aff_b = meta_b.get("affine", None)
76
assert_allclose(aff_a, aff_b)
77
- meta_a = {k: v for k, v in meta_a.items() if k != "affine"}
78
- meta_b = {k: v for k, v in meta_b.items() if k != "affine"}
+ meta_a = {k: v for k, v in meta_a.items() if k not in ("affine", "original_channel_dim")}
+ meta_b = {k: v for k, v in meta_b.items() if k not in ("affine", "original_channel_dim")}
79
self.assertEqual(meta_a, meta_b)
80
81
def check(
@@ -122,7 +122,7 @@ def test_as_tensor(self, device, dtype):
122
def test_as_dict(self):
123
m, _ = self.get_im()
124
m_dict = m.as_dict("im")
125
- im, meta = m_dict["im"], m_dict[PostFix.meta("im")]
+ im, meta = m_dict["im"], deepcopy(m_dict[PostFix.meta("im")])
126
affine = meta.pop("affine")
127
m2 = MetaTensor(im, affine, meta)
128
self.check(m2, m, check_ids=False)
0 commit comments