correct compilation errror related to generics

bioimage-io · Oct 1, 2024 · a646c1f · a646c1f
1 parent 1c8ea08
commit a646c1f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api030/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api030/Tensorflow2Interface.java
@@ -720,11 +720,11 @@ void main(String[] args) throws LoadModelException, IOException, RunModelExcepti
     	try {
         	pi.loadModel(modelFolder, modelSourc);
         	RandomAccessibleInterval<FloatType> rai = ArrayImgs.floats(new long[] {1, 512, 512, 1});
-        	Tensor<T> inp = (Tensor<T>) Tensor.build("aa", "byxc", rai);
-        	Tensor<R> out = (Tensor<R>) Tensor.buildBlankTensor("oo", "bcyx", new long[] {1, 512, 512, 33}, new FloatType());
+        	Tensor<FloatType> inp = Tensor.build("aa", "byxc", rai);
+        	Tensor<FloatType> out = Tensor.buildBlankTensor("oo", "bcyx", new long[] {1, 512, 512, 33}, new FloatType());
         	//Tensor<?> out = Tensor.buildEmptyTensor("oo", "byxc");
-        	List<Tensor<T>> ins = new ArrayList<Tensor<T>>();
-        	List<Tensor<R>> ous = new ArrayList<Tensor<R>>();
+        	List<Tensor<FloatType>> ins = new ArrayList<Tensor<FloatType>>();
+        	List<Tensor<FloatType>> ous = new ArrayList<Tensor<FloatType>>();
         	ins.add(inp);
         	ous.add(out);
         	pi.run(ins, ous);