def run_main():
"""Initializes flags and calls main()."""
program.setup_environment()
if getattr(tf, '__version__', 'stub') == 'stub':
print(
"TensorFlow installation not found - running with reduced feature set.",
file=sys.stderr)
tensorboard = program.TensorBoard(
default.get_plugins() + default.get_dynamic_plugins(),
program.get_default_assets_zip_provider(),
subcommands=[uploader_main.UploaderSubcommand()])
try:
from absl import app
# Import this to check that app.run() will accept the flags_parser argument.
from absl.flags import argparse_flags
app.run(tensorboard.main, flags_parser=tensorboard.configure)
raise AssertionError("absl.app.run() shouldn't return")
except ImportError:
pass
except base_plugin.FlagsError as e:
print("Error: %s" % e, file=sys.stderr)
sys.exit(1)
tensorboard.configure(sys.argv)
sys.exit(tensorboard.main())
def test_get_dynamic_plugin(self, mock_iter_entry_points):
mock_iter_entry_points.return_value = [FakeEntryPoint.create()]
actual_plugins = default.get_dynamic_plugins()
mock_iter_entry_points.assert_called_with('tensorboard_plugins')
self.assertEqual(actual_plugins, [FakePlugin])
def run(self):
tb = program.TensorBoard(
default.get_plugins() + default.get_dynamic_plugins(),
program.get_default_assets_zip_provider())
tb.configure(logdir = str(self.output / "tb"), port = self.port)
log = logging.getLogger('tensorflow')
log.setLevel(logging.CRITICAL)
tb.main()
def _plugins(disabled=None):
with warnings.catch_warnings():
warnings.simplefilter("ignore", Warning)
from tensorboard import default
base_plugins = default.get_plugins() + default.get_dynamic_plugins()
plugins = _filter_disabled_plugins(disabled, base_plugins)
log.debug("TensorBoard plugins: %s", plugins)
return plugins
def run(self):
# Remove http message
log = logging.getLogger('werkzeug').setLevel(logging.ERROR)
# Start tensorboard server
tb = customTensorBoard(
default.get_plugins() + default.get_dynamic_plugins(),
program.get_default_assets_zip_provider())
tb.configure(
argv=[None, '--logdir', self.dir_path, '--port',
str(self.port)])
url = tb.launch()
self.tb = tb
return url
def GatherAndServe(logdir: pathlib.Path, hosts: typing.List[str]):
paths = [str(logdir)] + [f"ssh://{host}{logdir}" for host in FLAGS.hosts]
with tempfile.TemporaryDirectory(prefix="phd_java_fuzz_") as d:
working_dir = pathlib.Path(d)
outdir = GatherAndArrangeLogs(paths, working_dir)
# Launch tensorboard
tensorboard = program.TensorBoard(
default.get_plugins() + default.get_dynamic_plugins(),
program.get_default_assets_zip_provider(),
)
tensorboard.configure(["unused_arg0", "--logdir", str(outdir)])
tensorboard_url = tensorboard.launch()
app.Log(1, "Serving tensorboard at %s", tensorboard_url)
while True:
GatherAndArrangeLogs(paths, working_dir)
time.sleep(10)
def base_plugins():
with warnings.catch_warnings():
warnings.simplefilter("ignore", Warning)
from tensorboard import default
return list(set(default.get_plugins() + default.get_dynamic_plugins()))
