From 36bce5717291aa627f265ddbda89e9e521638b4c Mon Sep 17 00:00:00 2001
From: Gagandeep Singh <gdp.1807@gmail.com>
Date: Thu, 12 Feb 2026 01:01:46 +0530
Subject: [PATCH] Update ml-metadata dependency to 1.17.1

- Update default version constraint in tfx/dependencies.py
- Update Bazel workspace tag in tfx/workspace.bzl
---
 tfx/dependencies.py | 2 +-
 tfx/workspace.bzl   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tfx/dependencies.py b/tfx/dependencies.py
index ca8469aefc..6c3cf88d6d 100644
--- a/tfx/dependencies.py
+++ b/tfx/dependencies.py
@@ -58,7 +58,7 @@ def make_pipeline_sdk_required_install_packages():
         "ml-metadata"
         + select_constraint(
             # LINT.IfChange
-            default=">=1.16.0,<1.17.0",
+            default=">=1.17.1,<1.18.0",
             # LINT.ThenChange(tfx/workspace.bzl)
             nightly=">=1.17.0.dev",
             git_master="@git+https://github.com/google/ml-metadata@master",
diff --git a/tfx/workspace.bzl b/tfx/workspace.bzl
index 19bb617e3d..19466fb9c3 100644
--- a/tfx/workspace.bzl
+++ b/tfx/workspace.bzl
@@ -79,7 +79,7 @@ def tfx_workspace():
         name = "com_github_google_ml_metadata",
         repo = "google/ml-metadata",
         # LINT.IfChange
-        tag = "v1.16.0",
+        tag = "v1.17.1",
         # LINT.ThenChange(//tfx/dependencies.py)
     )