diff --git a/tests/unit_tests/test_wandb_artifacts_cache.py b/tests/unit_tests/test_wandb_artifacts_cache.py index d7f46dcc9e6..f2aa16b4e12 100644 --- a/tests/unit_tests/test_wandb_artifacts_cache.py +++ b/tests/unit_tests/test_wandb_artifacts_cache.py @@ -2,7 +2,6 @@ import os import pathlib import random -import time from multiprocessing import Pool import pytest @@ -162,18 +161,22 @@ def test_artifacts_cache_cleanup(): os.makedirs(path_1) with open(os.path.join(path_1, "aardvark"), "w") as f: f.truncate(5000) - time.sleep(0.1) + f.flush() + os.fsync(f) path_2 = os.path.join(cache_root, "ab") os.makedirs(path_2) with open(os.path.join(path_2, "absolute"), "w") as f: f.truncate(2000) - time.sleep(0.1) + f.flush() + os.fsync(f) path_3 = os.path.join(cache_root, "ac") os.makedirs(path_3) with open(os.path.join(path_3, "accelerate"), "w") as f: f.truncate(1000) + f.flush() + os.fsync(f) cache = wandb_sdk.wandb_artifacts.ArtifactsCache("cache") reclaimed_bytes = cache.cleanup(5000)