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()))