Skip to content

Commit

Permalink
add utils
Browse files Browse the repository at this point in the history
  • Loading branch information
madhur-ob committed May 29, 2024
1 parent 53ee762 commit f2453be
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions metaflow/runner/utils.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
import os
import time
from typing import Dict


def clear_and_set_os_environ(env: Dict):
os.environ.clear()
os.environ.update(env)


def read_from_file_when_ready(file_path: str, timeout: float = 5):
start_time = time.time()
with open(file_path, "r", encoding="utf-8") as file_pointer:
content = file_pointer.read()
while not content:
if time.time() - start_time > timeout:
raise TimeoutError(
"Timeout while waiting for file content from '%s'" % file_path
)
time.sleep(0.1)
content = file_pointer.read()
return content

0 comments on commit f2453be

Please sign in to comment.