diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 82e95ec8c6..642b2f70e9 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -83,6 +83,10 @@ jobs: -r ./tensorbored/pip_package/requirements.txt \ -r ./tensorbored/pip_package/requirements_dev.txt \ ; + - name: 'Install Chrome dependencies' + run: | + sudo apt-get update + sudo apt-get install -y libgbm-dev libxss1 libasound2 - name: 'Check Pip state' run: pip freeze --all - name: 'Bazel: fetch' diff --git a/tensorbored/BUILD b/tensorbored/BUILD index 6e58dfdf02..19949b213d 100644 --- a/tensorbored/BUILD +++ b/tensorbored/BUILD @@ -1,12 +1,12 @@ # Description: # TensorBoard, a dashboard for investigating TensorFlow -load("//tensorbored/defs:py_repl.bzl", "py_repl") -load("//tensorbored/defs:web.bzl", "tf_web_library") -load("//tensorbored/defs:zipper.bzl", "tensorboard_zip_file") load("@rules_python//python:py_binary.bzl", "py_binary") load("@rules_python//python:py_library.bzl", "py_library") load("@rules_python//python:py_test.bzl", "py_test") +load("//tensorbored/defs:py_repl.bzl", "py_repl") +load("//tensorbored/defs:web.bzl", "tf_web_library") +load("//tensorbored/defs:zipper.bzl", "tensorboard_zip_file") package(default_visibility = [":internal"]) @@ -212,10 +212,13 @@ py_library( py_test( name = "manager_test", - size = "small", + size = "large", srcs = ["manager_test.py"], srcs_version = "PY3", - tags = ["support_notf"], + tags = [ + "exclusive", + "support_notf", + ], visibility = ["//tensorbored:internal"], deps = [ ":manager", @@ -274,10 +277,13 @@ py_library( py_test( name = "program_test", - size = "small", + size = "large", srcs = ["program_test.py"], srcs_version = "PY3", - tags = ["support_notf"], + tags = [ + "exclusive", + "support_notf", + ], deps = [ ":default", ":program", diff --git a/tensorbored/program_test.py b/tensorbored/program_test.py index c63fd54a27..b134c869ae 100644 --- a/tensorbored/program_test.py +++ b/tensorbored/program_test.py @@ -148,9 +148,10 @@ def testSpecifiedHost(self): ) self.assertStartsWith(server.get_url(), "http://127.0.0.1:") one_passed = True - except program.TensorBoardServerException: - # IPv4 is not supported + except (program.TensorBoardServerException, OSError, SystemExit): + # IPv4 is not supported or failed to bind pass + try: server = program.WerkzeugServer( self._StubApplication(), @@ -158,12 +159,14 @@ def testSpecifiedHost(self): ) self.assertStartsWith(server.get_url(), "http://[::1]:") one_passed = True - except program.TensorBoardServerException: - # IPv6 is not supported + except (program.TensorBoardServerException, OSError, SystemExit): + # IPv6 is not supported or failed to bind pass + self.assertTrue( - one_passed - ) # We expect either IPv4 or IPv6 to be supported + one_passed, + "Neither IPv4 (127.0.0.1) nor IPv6 (::1) could be bound.", + ) class SubcommandTest(tb_test.TestCase):