diff --git a/tests/test_infer/test_inference_engine.py b/tests/test_infer/test_inference_engine.py index edf76ba1b..b5f50baaa 100755 --- a/tests/test_infer/test_inference_engine.py +++ b/tests/test_infer/test_inference_engine.py @@ -44,7 +44,7 @@ def run_dist(rank, world_size, port): check_inference_engine(True) check_inference_engine(False) - # TODO: There are some in sampler + # TODO: There are some bugs in sampler. # for s1, s2 in zip(cai_outputs, transformer_outputs): # assert s1 == s2