Internal change

PiperOrigin-RevId: 543190939

Author: CLU Authors

clu/metric_writers/__init__.py

@@ -41,16 +41,15 @@
 """
 
 # 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,8 +19,6 @@
 method of the writer depending on the type of the metric.
 """
 
-# pylint: disable=g-importing-member
-
 import collections
 import getpass
 import os
@@ -31,10 +29,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,8 +13,6 @@
 # limitations under the License.
 
 """Tests for interface."""
-# pylint: disable=g-importing-member
-
 import itertools
 from typing import Any
 from unittest import mock
@@ -25,16 +23,17 @@
 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