Fix NPE in Signature.toString()

tensorflow · Jul 12, 2021 · 13cc616 · 13cc616
1 parent e0e4545
commit 13cc616
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Signature.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Signature.java
@@ -172,8 +172,9 @@ public Set<String> outputNames() {
   @Override
   public String toString() {
     StringBuilder strBuilder = new StringBuilder("Signature for \"" + key +"\":\n");
-    if (!methodName().isEmpty()) {
-      strBuilder.append("\tMethod: \"").append(methodName()).append("\"\n");
+    String methodName = methodName();
+    if (methodName != null && !methodName.isEmpty()) {
+      strBuilder.append("\tMethod: \"").append(methodName).append("\"\n");
     }
     if (signatureDef.getInputsCount() > 0) {
       strBuilder.append("\tInputs:\n");