diff --git a/tests/tt_eager/profiling/test_host_overhead_ci.py b/tests/tt_eager/profiling/test_host_overhead_ci.py index a0c64b1046d..4fd1b98d8bd 100644 --- a/tests/tt_eager/profiling/test_host_overhead_ci.py +++ b/tests/tt_eager/profiling/test_host_overhead_ci.py @@ -7,8 +7,10 @@ import csv import os import time +import pytest +@pytest.mark.timeout(10000) def test_host_overhead_ci(): profile_output_folder = "host_overhead_profile/" profile_output_filename = "final.csv"