Add tfrecord stub

elibixby · elibixby · commit 125dda05a123 · 2017-02-24T17:23:27.000-08:00
diff --git a/ml_engine/online_prediction/predict_test.py b/ml_engine/online_prediction/predict_test.py
@@ -34,3 +34,9 @@ def test_predict_json_error():
 def test_census_example_to_bytes():
     b = census_to_example_bytes(JSON)
     assert EXAMPLE_BYTE_STRING == base64.b64encode(b)
+
+def test_predict_tfrecord():
+    # Using the same model for TFRecords and
+    # JSON is currently broken.
+    # TODO(elibixby) when b/35742966 is fixed add
+    pass
