Skip to content

Commit 8759bc8

Browse files
author
Felipe Mello
committed
Merge branch 'timestamp_logging_diff1' into timestamp_logging_diff2
2 parents d82c354 + 4fe2611 commit 8759bc8

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/forge/controller/provisioner.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from forge.controller.launcher import BaseLauncher, get_launcher
2222

2323
from forge.env_constants import FORGE_DISABLE_METRICS
24-
from forge.observability.metric_actors import get_or_create_metric_logger
2524

2625
from forge.types import ProcessConfig, ProvisionerConfig
2726

@@ -265,6 +264,8 @@ def bootstrap(env: dict[str, str]):
265264

266265
# Spawn local fetcher actor on each process and register with global logger
267266
if os.getenv(FORGE_DISABLE_METRICS, "false").lower() != "true":
267+
from forge.observability.metric_actors import get_or_create_metric_logger
268+
268269
_ = await get_or_create_metric_logger(procs)
269270
return procs
270271

@@ -286,6 +287,10 @@ async def stop_proc_mesh(self, proc_mesh: ProcMesh):
286287
async with self._lock:
287288
# Deregister local logger from global logger
288289
if hasattr(proc_mesh, "_local_fetcher"):
290+
from forge.observability.metric_actors import (
291+
get_or_create_metric_logger,
292+
)
293+
289294
global_logger = await get_or_create_metric_logger(proc_mesh)
290295
await global_logger.deregister_fetcher.call_one(proc_mesh)
291296

0 commit comments

Comments
 (0)