diff --git a/exir/program/test/test_fake_program.py b/exir/program/test/test_fake_program.py index edccde43c2b..15959efde47 100644 --- a/exir/program/test/test_fake_program.py +++ b/exir/program/test/test_fake_program.py @@ -62,8 +62,8 @@ def test_fake_program(self) -> None: self.assertEqual(exported_program.verifier, fake_program.verifier) self.assertEqual(id(exported_program.verifier), id(fake_program.verifier)) - # Fake program uses fake tensors for the state dict. Size should be smaller. - self.assertLess( + # Fake program uses fake tensors for the state dict. Size should be not be larger. + self.assertLessEqual( sys.getsizeof(fake_program.state_dict), sys.getsizeof(exported_program.state_dict), )