Skip to content

Commit eaeda4c

Browse files
committed
Fix NPE in Signature.toString()
1 parent 762ae08 commit eaeda4c

File tree

1 file changed

+4
-3
lines changed
  • tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow

1 file changed

+4
-3
lines changed

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Signature.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -208,9 +208,10 @@ public Set<String> outputNames() {
208208

209209
@Override
210210
public String toString() {
211-
StringBuilder strBuilder = new StringBuilder("Signature for \"" + key + "\":\n");
212-
if (!methodName().isEmpty()) {
213-
strBuilder.append("\tMethod: \"").append(methodName()).append("\"\n");
211+
StringBuilder strBuilder = new StringBuilder("Signature for \"" + key +"\":\n");
212+
String methodName = methodName();
213+
if (methodName != null && !methodName.isEmpty()) {
214+
strBuilder.append("\tMethod: \"").append(methodName).append("\"\n");
214215
}
215216
if (signatureDef.getInputsCount() > 0) {
216217
strBuilder.append("\tInputs:\n");

0 commit comments

Comments
 (0)