From 300c74f25cfec1b1c9c668db973b0c90a0a5a7a0 Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Sun, 29 May 2022 18:10:12 +0200
Subject: [PATCH] refactor methods

---
 .../icy/tensorflow/v1/tensor/TensorBuilder.java                 | 2 +-
 src/main/java/tensorflow/v1/Tensorflow1Interface.java           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
index a8b9564..3332be6 100644
--- a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
+++ b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java
@@ -31,7 +31,7 @@ public final class TensorBuilder
      */
     public static Tensor<?> build(org.bioimageanalysis.icy.deeplearning.tensor.Tensor tensor)
     {
-    	return build(tensor.getDataAsNDArray());
+    	return build(tensor.getData());
     }
 
     /**
diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index c9ae2ab..ec43158 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -120,7 +120,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
 		if (outputNDArrays.size() != outputTensors.size())
 			throw new RunModelException(outputNDArrays.size(), outputTensors.size());
 		for (int i = 0; i < outputNDArrays.size(); i ++) {
-			outputTensors.get(i).setNDArrayData(Nd4fBuilder.build(outputNDArrays.get(i)));
+			outputTensors.get(i).setData(Nd4fBuilder.build(outputNDArrays.get(i)));
 		}
 		return outputTensors;
 	}
-- 
GitLab