Skip to content

Commit

Permalink
Add pillow, pyyaml, and requests to model_maker BUILD
Browse files Browse the repository at this point in the history
Fix setuptools=70.3.0 due to pypa/setuptools#4487

PiperOrigin-RevId: 683648416
  • Loading branch information
joezoug authored and copybara-github committed Oct 8, 2024
1 parent 2fe2076 commit b5d7b61
Show file tree
Hide file tree
Showing 5 changed files with 12 additions and 2 deletions.
5 changes: 4 additions & 1 deletion mediapipe/model_maker/python/core/data/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,10 @@ py_test(
py_library(
name = "cache_files",
srcs = ["cache_files.py"],
deps = ["@model_maker_pip_deps_tensorflow//:pkg"],
deps = [
"@model_maker_pip_deps_pyyaml//:pkg",
"@model_maker_pip_deps_tensorflow//:pkg",
],
)

py_test(
Expand Down
4 changes: 4 additions & 0 deletions mediapipe/model_maker/python/core/utils/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,9 @@ py_test(
py_library(
name = "file_util",
srcs = ["file_util.py"],
deps = [
"@model_maker_pip_deps_requests//:pkg",
],
)

py_test(
Expand All @@ -74,6 +77,7 @@ py_test(
deps = [
":file_util",
"@model_maker_pip_deps_absl_py//:pkg",
"@model_maker_pip_deps_requests//:pkg",
],
)

Expand Down
1 change: 1 addition & 0 deletions mediapipe/model_maker/python/vision/core/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ py_library(
name = "image_utils",
srcs = ["image_utils.py"],
deps = [
"@model_maker_pip_deps_pillow//:pkg",
"@model_maker_pip_deps_tensorflow//:pkg",
],
)
Expand Down
1 change: 1 addition & 0 deletions mediapipe/model_maker/requirements_bazel.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
absl-py
numpy<2
opencv-python
setuptools==70.3.0 # needed due to https://github.com/pypa/setuptools/issues/4487
tensorflow>=2.10,<2.16
tensorflow-addons
tensorflow-datasets
Expand Down
3 changes: 2 additions & 1 deletion mediapipe/model_maker/requirements_lock.txt
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,8 @@ zipp==3.20.2
# importlib-resources

# The following packages are considered to be unsafe in a requirements file:
setuptools==75.1.0
setuptools==70.3.0
# via
# -r mediapipe/opensource_only/model_maker_requirements_bazel.txt
# tensorboard
# tensorflow

0 comments on commit b5d7b61

Please sign in to comment.