Resolve few export issues with nightly releases.

They are still failing during smoke tests for TF compatibility because TF made a new release with likely incompatible protobuf version, so we'd have to update protobuf dependency if there's enough users wanting to upgrade to the new TF.

PiperOrigin-RevId: 887310852

Author: iindyk

grain/_src/python/checkpoint/handler_test.py

@@ -29,10 +29,6 @@ py_library(
     name = "tfrecord_dataset",
     srcs = ["tfrecord_dataset.py"],
     srcs_version = "PY3",
-    visibility = [
-        "//third_party/py/grain:internal",
-        "//third_party/py/maxtext:__pkg__",
-    ],
     deps = ["//grain/_src/python/dataset"],
 )
 

grain/_src/python/checkpoint/orbax_integration_test.py

@@ -21,7 +21,6 @@ main() {
   # Enable host OS specific configs. For instance, "build:linux" will be used
   # automatically when building on Linux.
   write_to_bazelrc "build --enable_platform_specific_config"
-  write_to_bazelrc "build --verbose_failures"
   # Bazel 7.0.0 no longer supports dynamic symbol lookup on macOS. To resolve
   # undefined symbol errors in macOS arm64 builds, explicitly add the necessary
   # linker flags until dependencies are well defined. See