Project import generated by Copybara.

PiperOrigin-RevId: 210432162

README.md

@@ -57,7 +57,8 @@ combinations may also work.
 
 |tensorflow-model-analysis  |tensorflow    |apache-beam[gcp]|
 |---------------------------|--------------|----------------|
-|GitHub master              |1.9           |2.5.0           |
+|GitHub master              |1.9           |2.6.0           |
+|0.9.1                      |1.9           |2.6.0           |
 |0.9.0                      |1.9           |2.5.0           |
 |0.6.0                      |1.6           |2.4.0           |
 

RELEASE.md

@@ -1,3 +1,22 @@
+# Release 0.9.1
+
+## Major Features and Improvements
+
+## Bug fixes and other changes
+
+*   Requires pre-installed TensorFlow >=1.10,<2.
+*   Updated ExampleCount to use the batch dimension as the example count. It
+    also now tries a few fallbacks if none of the standard keys are found in the
+    predictions dictionary: the first key in sorted order in the predictions
+    dictionary, or failing that, the first key in sorted order in the labels
+    dictionary, or failing that, it defaults to zero.
+*   Fix bug where we were mutating an element in a DoFn - this is prohibited in
+    the Beam model and can cause subtle bugs.
+
+## Breaking changes
+
+## Deprecations
+
 # Release 0.9.0
 
 ## Major Features and Improvements

examples/chicago_taxi/README.md

@@ -45,8 +45,8 @@ Development for this example will be isolated in a Python virtual environment.
 This allows us to experiment with different versions of dependencies.
 
 There are many ways to install `virtualenv`, see the
-[TensorFlow install guides](/install) for different platforms, but here are a
-couple:
+[TensorFlow install guides](https://www.tensorflow.org/install) for different
+platforms, but here are a couple:
 
 * For Linux:
 

setup.py

@@ -243,7 +243,7 @@ def run(self):
         'tensorflow_model_analysis/static/vulcanized_template.html',
     ]),],
     'install_requires': [
-        'apache-beam[gcp]>=2.5,<3',
+        'apache-beam[gcp]>=2.6,<3',
         'grpc-google-iam-v1==0.11.1',
         'numpy>=1.10,<2',
         'jupyter>=1.0,<2',

tensorflow_model_analysis/BUILD

@@ -1,10 +1,30 @@
 # Description:
 #   Python SDK for the TensorFlow Model Analysis API.
 
+package(default_visibility = ["//tensorflow_model_analysis:__subpackages__"])
+
 licenses(["notice"])  # Apache 2.0
 
 exports_files(["LICENSE"])
 
+py_library(
+    name = "tensorflow_model_analysis",
+    srcs = [
+        "__init__.py",
+    ],
+    visibility = ["//visibility:public"],
+    deps = [
+        ":constants",
+        ":version",
+        "//tensorflow_model_analysis/api:model_eval_lib",
+        "//tensorflow_model_analysis/api:tfma_unit_non_testonly",
+        "//tensorflow_model_analysis/eval_saved_model:export",
+        "//tensorflow_model_analysis/eval_saved_model:exporter",
+        "//tensorflow_model_analysis/eval_saved_model/post_export_metrics",
+        "//tensorflow_model_analysis/slicer",
+    ],
+)
+
 py_library(
     name = "constants",
     srcs = ["constants.py"],