Skip to content

Commit 2ce88bf

Browse files
andsteingcopybara-github
authored andcommitted
Internal change
PiperOrigin-RevId: 529646048
1 parent 9cfdeed commit 2ce88bf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

clu/metric_writers/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
"""
2121

2222
import collections
23+
import getpass
2324
import os
2425
import re
2526
from typing import Any, List, Mapping, Optional, Tuple, Union
@@ -97,6 +98,8 @@ def write_values(writer: MetricWriter, step: int,
9798
fn(step, vals, **dict(extra_args))
9899

99100

101+
102+
100103
def create_default_writer(
101104
logdir: Optional[epath.PathLike] = None,
102105
*,
@@ -114,7 +117,6 @@ def create_default_writer(
114117
just_logging: If True only use a LoggingWriter. This is useful in multi-host
115118
setups when only the first host should write metrics and all other hosts
116119
should only write to their own logs.
117-
write_to_xm_measurements: If True uses XmMeasurementsWriter in addition.
118120
default (None) will automatically determine if you # GOOGLE-INTERNAL have
119121
asynchronous: If True return an AsyncMultiWriter to not block when writing
120122
metrics.

0 commit comments

Comments
 (0)