Skip to content

Commit 9dbf608

Browse files
committed
Bug fix
1 parent 6065388 commit 9dbf608

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tfliteiorewriter/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def rewrite(
103103
flat_json = json.load(f)
104104

105105
# Checks if signature_defs are recorded in tflite
106-
if 'signature_defs' not in flat_json:
106+
if 'signature_defs' not in flat_json or not flat_json['signature_defs']:
107107
print(
108108
f'{Color.YELLOW}WARNING:{Color.RESET} ' +
109109
f'Processing is aborted because signature_defs is not recorded in tflite.'

0 commit comments

Comments
 (0)