make the types more specific

bioimage-io · Sep 23, 2024 · 57d1dda · 57d1dda
1 parent c32a330
commit 57d1dda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/engine/DeepLearningEngineInterface.java b/src/main/java/io/bioimage/modelrunner/engine/DeepLearningEngineInterface.java
@@ -22,7 +22,8 @@
 import java.util.List;
 
 import io.bioimage.modelrunner.tensor.Tensor;
-
+import net.imglib2.type.NativeType;
+import net.imglib2.type.numeric.RealType;
 import io.bioimage.modelrunner.exceptions.LoadModelException;
 import io.bioimage.modelrunner.exceptions.RunModelException;
 
@@ -47,7 +48,8 @@ public interface DeepLearningEngineInterface
 	 * TODO should it be like the commented option? ask jean-yves
 	 */
 	//public void run( List< Tensor > inputTensors, List< Tensor > outputTensors ) throws RunModelException;
-	public void run( List< Tensor  < ? > > inputTensors, List< Tensor  < ? > > outputTensors ) throws RunModelException;
+	public <T extends RealType<T> & NativeType<T>, R extends RealType<R> & NativeType<R>>
+	void run( List< Tensor  < T > > inputTensors, List< Tensor  < R > > outputTensors ) throws RunModelException;
 
 	/**
 	 * Load the model with the corresponding engine on the particular