From 2e2c0f1ce9b4f991d8f2cb40457da999ba63ad32 Mon Sep 17 00:00:00 2001 From: Marvin Ritter Date: Fri, 23 Jun 2023 00:09:57 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 542778597 --- clu/metric_writers/__init__.py | 3 ++- clu/metric_writers/utils.py | 4 +++- clu/metric_writers/utils_test.py | 5 +++-- 3 files changed, 8 insertions(+), 4 deletions(-) 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