Commit 8186ecd2 authored by Gaetan Lepage's avatar Gaetan Lepage Committed by Florian Klink
Browse files

python311Packages.tensorflow-metadata: cleanup

parent 84e9ff03
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
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(
+6 −1
Original line number Diff line number Diff line
@@ -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 ];
  };