Skip to content

Commit ad9172c

Browse files
committed
Missed a bit in the refactor.
1 parent b122dd0 commit ad9172c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@ public List<SessionFunction> functions() {
468468
* @return list of output tensors, mapped by the signature name
469469
* @throws IllegalArgumentException if no function can be selected by default
470470
*/
471-
public Map<String, Tensor> call(Map<String, Tensor> arguments) {
471+
public Result call(Map<String, Tensor> arguments) {
472472
SessionFunction function = null;
473473
if (functions.size() == 1) {
474474
function = functions.values().iterator().next();

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ public void exportFunctionWithVariables() throws IOException {
215215
// Now call the same function directly from the model
216216
try (TFloat32 zTensor =
217217
(TFloat32)
218-
savedModel.call(Collections.singletonMap("input", xTensor)).get("reducedSum")) {
218+
savedModel.call(Collections.singletonMap("input", xTensor)).get("reducedSum").get()) {
219219
assertEquals(reducedSum, zTensor.getFloat(), EPSILON);
220220
}
221221
}

0 commit comments

Comments
 (0)