Skip to content

Commit e7f81c2

Browse files
MarkDaoustcopybara-github
authored andcommitted
Add support for protobuf versions before 3.20
Copied the previous version of the generated file. Use the previous version if the protobuf version is before 3.20 PiperOrigin-RevId: 517006172
1 parent 39ff245 commit e7f81c2

File tree

3 files changed

+512
-2
lines changed

3 files changed

+512
-2
lines changed

site/en/tutorials/audio/music_generation.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1081,7 +1081,7 @@
10811081
" # Add batch dimension\n",
10821082
" inputs = tf.expand_dims(notes, 0)\n",
10831083
"\n",
1084-
" predictions = keras_model.predict(inputs)\n",
1084+
" predictions = model.predict(inputs)\n",
10851085
" pitch_logits = predictions['pitch']\n",
10861086
" step = predictions['step']\n",
10871087
" duration = predictions['duration']\n",

tools/tensorflow_docs/api_generator/report/schema/__init__.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,18 @@
1414
# ==============================================================================
1515
"""Public API for api report proto."""
1616

17-
from tensorflow_docs.api_generator.report.schema import api_report_generated_pb2 as api_report_pb2
17+
import sys
18+
19+
from google import protobuf
20+
21+
_version_parts = protobuf.__version__.split('.')
22+
_version = (int(_version_parts[0]), int(_version_parts[1]))
23+
24+
if _version >= (3, 20):
25+
from tensorflow_docs.api_generator.report.schema import api_report_generated_pb2 as api_report_pb2 # pylint: disable=g-import-not-at-top
26+
else:
27+
from tensorflow_docs.api_generator.report.schema import api_report_generated_319_pb2 as api_report_pb2 # pylint: disable=g-import-not-at-top
28+
29+
sys.modules['tensorflow_docs.api_generator.report.schema.api_report_pb2'] = (
30+
api_report_pb2
31+
)

0 commit comments

Comments
 (0)