fix: Replace flaky with pytest.mark.flaky

fix: Replace flaky with pytest.mark.flaky #2237