def _patch_argparse(output_path):
handle_parse = lambda parse_args, *_args, **_kw: _write_flags(
parse_args, output_path)
# parse_known_args is called by parse_args, so this handles both
# cases.
python_util.listen_method(argparse.ArgumentParser, "parse_known_args",
handle_parse)
def ___patch_argparse(output_path):
___python_util.listen_method(___argparse.ArgumentParser, "add_argument",
___handle_add_argument)
___handle_parse = lambda *args, **kw: ___write_flags_and_exit(output_path)
# parse_known_args is called by parse_args, so this handled both
# cases.
___python_util.listen_method(___argparse.ArgumentParser,
"parse_known_args", ___handle_parse)
def _patch_tensorboardX(self):
try:
from tensorboardX import SummaryWriter
except ImportError:
pass
else:
self.log.debug("wrapping tensorboardX.SummaryWriter.add_scalar")
python_util.listen_method(SummaryWriter, "add_scalar", self._handle_scalar)
def _try_listen_tf_legacy(self):
if not _tf_version().startswith("1."):
raise util.TryFailed()
try:
# pylint: disable=import-error,no-name-in-module
from tensorflow.summary import FileWriter
except Exception as e:
self.log.debug("error importing tensorflow.summary.FileWriter: %s", e)
raise util.TryFailed()
else:
self.log.debug("wrapping tensorflow.summary.FileWriter.add_summary")
python_util.listen_method(FileWriter, "add_summary", self._handle_summary)
def _try_patch_tensorflow(self):
try:
import tensorflow
except ImportError:
self.log.debug(
"tensorflow cannot be imported - skipping summaries for %s",
self.name)
else:
self.log.debug(
"wrapping tensorflow.summary.FileWriter.add_summary")
python_util.listen_method(tensorflow.summary.FileWriter,
"add_summary", self._handle_summary)
def _try_listen_tf_v1(self):
if not _tf_version().startswith("1."):
raise util.TryFailed()
try:
from tensorflow.compat.v1.summary import FileWriter
except Exception as e:
self.log.debug(
"error importing tensorflow.compat.v1.summary.FileWriter: %s",
e)
raise util.TryFailed()
else:
self.log.debug(
"wrapping tensorflow.compat.v1.summary.FileWriter.add_summary")
python_util.listen_method(FileWriter, "add_summary",
self._handle_summary)
def _patch_guild_summary(self):
from guild import summary
self.log.debug("wrapping guild.summary.SummaryWriter.add_scalar")
wrapper = python_util.listen_method(
summary.SummaryWriter, "add_scalar", self._handle_guild_scalar
)
return wrapper # Return value used for tests.
def _try_patch_tensorboardX(self):
try:
from tensorboardX import SummaryWriter
except ImportError:
pass
else:
self.log.debug("wrapping tensorboardX.SummaryWriter.add_scalar")
wrapper = python_util.listen_method(
SummaryWriter, "add_scalar", self._handle_tbx_scalar
)
return wrapper # Return value used for tests.
def _patch_guild_summary(self):
from guild import summary
python_util.listen_method(summary.SummaryWriter, "add_scalar",
self._handle_guild_scalar)
def _patch_click(base_args, output_path):
handle_command_call = lambda group_call, *_args, **_kw: _write_command_flags(
group_call, base_args, output_path)
python_util.listen_method(click.Command, "__call__", handle_command_call)
def _patch_click(output_path):
handle_cmd_init = lambda _init, *_args, **kw: _write_flags(
kw["params"], output_path
)
python_util.listen_method(click.Command, "__init__", handle_cmd_init)
