diff options
Diffstat (limited to 'tools/tflitefile_tool/tflite/SkipGramOptions.py')
-rw-r--r-- | tools/tflitefile_tool/tflite/SkipGramOptions.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tools/tflitefile_tool/tflite/SkipGramOptions.py b/tools/tflitefile_tool/tflite/SkipGramOptions.py index 50738b924..9eb5059ea 100644 --- a/tools/tflitefile_tool/tflite/SkipGramOptions.py +++ b/tools/tflitefile_tool/tflite/SkipGramOptions.py @@ -37,8 +37,9 @@ class SkipGramOptions(object): def IncludeAllNgrams(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) if o != 0: - return self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos) - return 0 + return bool( + self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) + return False def SkipGramOptionsStart(builder): |