Fix for empty SavedModelBundle tags. (#611)

Author: Craigacp

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

@@ -312,6 +312,25 @@ public void export() throws IOException {
     private final Map<String, SessionFunction> functions = new LinkedHashMap<>();
   }
 
+  /**
+   * Load a saved model from an export directory. The model that is being loaded should be created
+   * using the <a href="https://www.tensorflow.org/api_docs/python/tf/saved_model">Saved Model
+   * API</a>.
+   *
+   * <p>This method is a shorthand for:
+   *
+   * <pre>{@code
+   * SavedModelBundle.loader().load();
+   * }</pre>
+   *
+   * @param exportDir the directory path containing a saved model.
+   * @return a bundle containing the graph and associated session.
+   */
+  public static SavedModelBundle load(String exportDir) {
+    Loader loader = loader(exportDir);
+    return loader.load();
+  }
+
   /**
    * Load a saved model from an export directory. The model that is being loaded should be created
    * using the <a href="https://www.tensorflow.org/api_docs/python/tf/saved_model">Saved Model
@@ -329,9 +348,7 @@ public void export() throws IOException {
    */
   public static SavedModelBundle load(String exportDir, String... tags) {
     Loader loader = loader(exportDir);
-    if (tags != null && tags.length > 0) {
-      loader.withTags(tags);
-    }
+    loader.withTags(tags);
     return loader.load();
   }
 

tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SavedModelBundleTest.java

@@ -68,6 +68,11 @@ public class SavedModelBundleTest {
 
   @Test
   public void load() {
+    try (SavedModelBundle bundle = SavedModelBundle.load(SAVED_MODEL_PATH)) {
+      assertNotNull(bundle.session());
+      assertNotNull(bundle.graph());
+      assertNotNull(bundle.metaGraphDef());
+    }
     try (SavedModelBundle bundle = SavedModelBundle.load(SAVED_MODEL_PATH, "serve")) {
       assertNotNull(bundle.session());
       assertNotNull(bundle.graph());