python311Packages.tensorflow-metadata: cleanup

This commit is contained in:
Gaetan Lepage
2025-04-23 18:47:16 +02:00
committed by Florian Klink
parent 84e9ff0353
commit 8186ecd283
2 changed files with 7 additions and 2 deletions
@@ -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(
@@ -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 ];
};