Assertion error in test_pass_annotation

Hello, I was looking at up to date examples for device annotation and when I tried to execute test_pass_annotation.py it showed me this assertion error:

Traceback (most recent call last):
  File "heterogenous exec.py", line 608, in <module>
    test_check_run()
  File "heterogenous exec.py", line 564, in test_check_run
    run_unpropagatable_graph(dev, tgt)
  File "heterogenous exec.py", line 549, in run_unpropagatable_graph
    assert device_index == expected_index
AssertionError

Indeed device_index is [4, 4, 4, 1, 1, 1, 4, 4] instead of [2, 2, 2, 1, 1, 1, 2, 2]. I does not look like a big bug but it still bothers me that a test fails. Was this error to be expected or is this a bug? I am using the latest version on the master branch.

Hi Experts, I am also facing same issue. Please kindly help on this problem.

Thanks in advance