Internal change.

PiperOrigin-RevId: 543185453

Author: Marvin182

clu/metric_writers/__init__.py

@@ -41,15 +41,16 @@
 """
 
 # pylint: disable=unused-import
+# pylint: disable=g-importing-member
 
 
 from clu.metric_writers.async_writer import AsyncMultiWriter
 from clu.metric_writers.async_writer import AsyncWriter
 from clu.metric_writers.async_writer import ensure_flushes
+from clu.metric_writers.summary_writer import SummaryWriter
 from clu.metric_writers.interface import MetricWriter
 from clu.metric_writers.logging_writer import LoggingWriter
 from clu.metric_writers.multi_writer import MultiWriter
-from clu.metric_writers.summary_writer import SummaryWriter
 from clu.metric_writers.utils import create_default_writer
 from clu.metric_writers.utils import write_values
 

clu/metric_writers/utils.py

@@ -19,6 +19,8 @@
 method of the writer depending on the type of the metric.
 """
 
+# pylint: disable=g-importing-member
+
 import collections
 import getpass
 import os
@@ -29,10 +31,10 @@
 from absl import logging
 from clu import values
 from clu.metric_writers.async_writer import AsyncMultiWriter
+from clu.metric_writers.summary_writer import SummaryWriter
 from clu.metric_writers.interface import MetricWriter
 from clu.metric_writers.logging_writer import LoggingWriter
 from clu.metric_writers.multi_writer import MultiWriter
-from clu.metric_writers.summary_writer import SummaryWriter
 from etils import epath
 import jax.numpy as jnp
 import numpy as np

clu/metric_writers/utils_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 
 """Tests for interface."""
+# pylint: disable=g-importing-member
+
 import itertools
 from typing import Any
 from unittest import mock
@@ -23,17 +25,16 @@
 from clu.metric_writers import utils
 from clu.metric_writers.async_writer import AsyncMultiWriter
 from clu.metric_writers.async_writer import AsyncWriter
+from clu.metric_writers.summary_writer import SummaryWriter
 from clu.metric_writers.interface import MetricWriter
 from clu.metric_writers.logging_writer import LoggingWriter
 from clu.metric_writers.multi_writer import MultiWriter
-from clu.metric_writers.summary_writer import SummaryWriter
 import clu.metrics
 import flax.struct
 import jax.numpy as jnp
 import tensorflow as tf
 
 
-
 @flax.struct.dataclass
 class HistogramMetric(clu.metrics.Metric):
   value: jnp.ndarray