add predictions_to_bigquery

tensorflow · May 23, 2023 · 71ddaf4 · 71ddaf4
1 parent 8507648
commit 71ddaf4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tfx_addons/version.py b/tfx_addons/version.py
@@ -79,5 +79,6 @@
     "huggingface_pusher":
     [f"tfx{_TFXVERSION_CONSTRAINT}", "huggingface-hub>=0.10.0,<1.0.0"],
     "model_card_generator":
-    [f"tfx{_TFXVERSION_CONSTRAINT}", "model-card-toolkit>=2.0.0,<3.0.0"]
+    [f"tfx{_TFXVERSION_CONSTRAINT}", "model-card-toolkit>=2.0.0,<3.0.0"],
+    "predictions_to_bigquery": [f"tfx{_TFXVERSION_CONSTRAINT}"]
 }