From 8186ecd283bf2afab1b887ac196b4c7394fcfa60 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 16 Oct 2024 09:33:34 +0200
Subject: [PATCH] python311Packages.tensorflow-metadata: cleanup

---
 .../python-modules/tensorflow-metadata/build.patch         | 2 +-
 .../python-modules/tensorflow-metadata/default.nix         | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/tensorflow-metadata/build.patch b/pkgs/development/python-modules/tensorflow-metadata/build.patch
index 75a774535947..14469d1f60fa 100644
--- a/pkgs/development/python-modules/tensorflow-metadata/build.patch
+++ b/pkgs/development/python-modules/tensorflow-metadata/build.patch
@@ -1,5 +1,5 @@
 diff --git a/setup.py b/setup.py
-index 7a09b2f..94c5aa6 100644
+index 30ac370..a05812d 100644
 --- a/setup.py
 +++ b/setup.py
 @@ -141,8 +141,4 @@ setup(
diff --git a/pkgs/development/python-modules/tensorflow-metadata/default.nix b/pkgs/development/python-modules/tensorflow-metadata/default.nix
index b961578c8810..c1ccd3c3befd 100644
--- a/pkgs/development/python-modules/tensorflow-metadata/default.nix
+++ b/pkgs/development/python-modules/tensorflow-metadata/default.nix
@@ -45,11 +45,16 @@ buildPythonPackage rec {
   # has no tests
   doCheck = false;
 
-  pythonImportsCheck = [ "tensorflow_metadata" ];
+  pythonImportsCheck = [
+    "tensorflow_metadata"
+    "tensorflow_metadata.proto.v0"
+    "google.protobuf.runtime_version"
+  ];
 
   meta = {
     description = "Standard representations for metadata that are useful when training machine learning models with TensorFlow";
     homepage = "https://github.com/tensorflow/metadata";
+    changelog = "https://github.com/tensorflow/metadata/releases/tag/v${version}";
     license = lib.licenses.asl20;
     maintainers = with lib.maintainers; [ ndl ];
   };