Skip to content

Commit e36da23

Browse files
committed
update to new version of JDLL
1 parent 3fbe40d commit e36da23

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,7 @@ private List<String> encodeInputs(List<Tensor<?>> inputTensors) throws FileAlrea
415415
HashMap<String, Object> map = new HashMap<String, Object>();
416416
map.put(NAME_KEY, tt.getName());
417417
map.put(SHAPE_KEY, tt.getShape());
418-
map.put(DTYPE_KEY, CommonUtils.getDataType(tt.getData()));
418+
map.put(DTYPE_KEY, CommonUtils.getDataTypeFromRAI(tt.getData()));
419419
map.put(IS_INPUT_KEY, true);
420420
map.put(MEM_NAME_KEY, shma.getName());
421421
encodedInputTensors.add(gson.toJson(map));
@@ -433,7 +433,7 @@ private List<String> encodeOutputs(List<Tensor<?>> outputTensors) throws FileAlr
433433
map.put(IS_INPUT_KEY, false);
434434
if (!tt.isEmpty()) {
435435
map.put(SHAPE_KEY, tt.getShape());
436-
map.put(DTYPE_KEY, CommonUtils.getDataType(tt.getData()));
436+
map.put(DTYPE_KEY, CommonUtils.getDataTypeFromRAI(tt.getData()));
437437
SharedMemoryArray shma = SharedMemoryArray.createSHMAFromRAI(tt.getData(), false, true);
438438
shmaOutputList.add(shma);
439439
map.put(MEM_NAME_KEY, shma.getName());

0 commit comments

Comments
 (0)