diff --git a/clu/metric_writers/__init__.py b/clu/metric_writers/__init__.py index 356e639..7232bff 100644 --- a/clu/metric_writers/__init__.py +++ b/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 diff --git a/clu/metric_writers/utils.py b/clu/metric_writers/utils.py index f02ad01..c4fe38e 100644 --- a/clu/metric_writers/utils.py +++ b/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 diff --git a/clu/metric_writers/utils_test.py b/clu/metric_writers/utils_test.py index 53e1831..eda6cef 100644 --- a/clu/metric_writers/utils_test.py +++ b/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