From ef6aeb88285152b88287c3a4db44a71d9ce77eb8 Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Tue, 20 Jun 2023 14:36:55 -0700 Subject: [PATCH] Allow passing of HParams to MediaPipe training docker PiperOrigin-RevId: 542052304 --- mediapipe/model_maker/python/core/BUILD | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mediapipe/model_maker/python/core/BUILD b/mediapipe/model_maker/python/core/BUILD index 0ed20a2fe..a73e545d3 100644 --- a/mediapipe/model_maker/python/core/BUILD +++ b/mediapipe/model_maker/python/core/BUILD @@ -14,7 +14,10 @@ # Placeholder for internal Python strict library and test compatibility macro. -package(default_visibility = ["//mediapipe:__subpackages__"]) +package(default_visibility = [ + "//cloud/ml/applications/vision/model_garden/model_oss/mediapipe:__subpackages__", + "//mediapipe:__subpackages__", +]) licenses(["notice"])