def _listen_tf_v2_summary(self):
with warnings.catch_warnings():
warnings.simplefilter("ignore", Warning)
from tensorboard.plugins.scalar import summary_v2
self.log.debug(
"wrapping tensorboard.plugins.scalar.summary_v2.scalar")
python_util.listen_function(summary_v2, "scalar", self._handle_scalar)
def _listen_tf_v2_summary(self):
with warnings.catch_warnings():
warnings.simplefilter("ignore", Warning)
# pylint: disable=import-error,no-name-in-module
from tensorflow.python.ops import summary_ops_v2
self.log.debug("wrapping tensorflow.python.ops summary_ops_v2.scalar")
python_util.listen_function(summary_ops_v2, "scalar",
self._handle_scalar_ops_v2)
def _listen_tf_summary(self):
# pylint: disable=import-error,no-name-in-module
from tensorflow import summary
self.log.debug("wrapping tensorflow.summary.scalar")
python_util.listen_function(summary, "scalar", self._handle_scalar)
def _listen_tf_v2_summary(self):
from tensorboard.plugins.scalar import summary_v2
self.log.debug("wrapping tensorboard.plugins.scalar.summary_v2.scalar")
python_util.listen_function(summary_v2, "scalar", self._handle_scalar)
def _patch():
from guild import op
from guild import python_util
python_util.listen_function(op, "run", _patched_op_run)
def _listen_tf_summary(self):
from tensorflow import summary
self.log.debug("wrapping tensorflow.summary.scalar")
python_util.listen_function(summary, "scalar", self._handle_scalar)