diff --git a/pkgs/development/python-modules/tensorflow-metadata/build.patch b/pkgs/development/python-modules/tensorflow-metadata/build.patch
index 14469d1f60fa..9cdf63d1434d 100644
--- a/pkgs/development/python-modules/tensorflow-metadata/build.patch
+++ b/pkgs/development/python-modules/tensorflow-metadata/build.patch
@@ -1,13 +1,13 @@
 diff --git a/setup.py b/setup.py
-index 30ac370..a05812d 100644
+index 9c2a17f..c7e120d 100644
 --- a/setup.py
 +++ b/setup.py
-@@ -141,8 +141,4 @@ setup(
-     keywords='tensorflow metadata tfx',
-     download_url='https://github.com/tensorflow/metadata/tags',
+@@ -154,8 +154,4 @@ setup(
+     keywords="tensorflow metadata tfx",
+     download_url="https://github.com/tensorflow/metadata/tags",
      requires=[],
 -    cmdclass={
--        'build': _BuildCommand,
--        'bazel_build': _BazelBuildCommand,
+-        "build": _BuildCommand,
+-        "bazel_build": _BazelBuildCommand,
 -    },
  )
diff --git a/pkgs/development/python-modules/tensorflow-metadata/default.nix b/pkgs/development/python-modules/tensorflow-metadata/default.nix
index c9cd7fae87f3..a73c766364cc 100644
--- a/pkgs/development/python-modules/tensorflow-metadata/default.nix
+++ b/pkgs/development/python-modules/tensorflow-metadata/default.nix
@@ -14,18 +14,24 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "tensorflow-metadata";
-  version = "1.17.3";
+  version = "1.21.0";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "tensorflow";
     repo = "metadata";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-4xj2JfQryGy7a9ho9/JJtyg+0H5VLQzq9JevOmOWJZk=";
+    hash = "sha256-k+SJmR5n4S31wpSuMhJwrjJfX/Bow0QwLpw+TwRPS7U=";
   };
 
   patches = [ ./build.patch ];
 
+  postPatch = ''
+    substituteInPlace pyproject.toml \
+      --replace-fail "setuptools<70" "setuptools"
+  '';
+
   # Default build pulls in Bazel + extra deps, given the actual build
   # is literally three lines (see below) - replace it with custom build.
   preBuild = ''