Skip to content

Commit aa1bcdd

Browse files
committed
fixes tests
Signed-off-by: Wenqi Li <[email protected]>
1 parent 0bf0efa commit aa1bcdd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_meta_tensor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,8 @@ def check_meta(self, a: MetaTensor, b: MetaTensor) -> None:
7474
aff_a = meta_a.get("affine", None)
7575
aff_b = meta_b.get("affine", None)
7676
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"}
77+
meta_a = {k: v for k, v in meta_a.items() if k not in ("affine", "original_channel_dim")}
78+
meta_b = {k: v for k, v in meta_b.items() if k not in ("affine", "original_channel_dim")}
7979
self.assertEqual(meta_a, meta_b)
8080

8181
def check(
@@ -122,7 +122,7 @@ def test_as_tensor(self, device, dtype):
122122
def test_as_dict(self):
123123
m, _ = self.get_im()
124124
m_dict = m.as_dict("im")
125-
im, meta = m_dict["im"], m_dict[PostFix.meta("im")]
125+
im, meta = m_dict["im"], deepcopy(m_dict[PostFix.meta("im")])
126126
affine = meta.pop("affine")
127127
m2 = MetaTensor(im, affine, meta)
128128
self.check(m2, m, check_ids=False)

0 commit comments

Comments
 (0)