Fix issue with mismatched int dtype in proto

Modular-ML-inference · Jan 1, 2024 · 0ee07ac · 0ee07ac
1 parent 7a665d6
commit 0ee07ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/main/scala/eu/assistiot/inference/benchmark/stream/encoding/FallEncodingFlow.scala b/src/main/scala/eu/assistiot/inference/benchmark/stream/encoding/FallEncodingFlow.scala
@@ -30,7 +30,7 @@ object FallEncodingFlow extends MlEncodingFlow[FallInferenceInput, FallInference
           id = input.id,
           input = Map(
             "array" -> TensorProto(
-              dtype = DataType.DT_INT64,
+              dtype = DataType.DT_INT16,
               tensorShape = shape,
               intVal = flatTensor,
             )

diff --git a/src/main/scala/eu/assistiot/inference/benchmark/util/Util.scala b/src/main/scala/eu/assistiot/inference/benchmark/util/Util.scala
@@ -0,0 +1,9 @@
+package eu.assistiot.inference.benchmark.util
+
+object Util:
+  def bytesToHex(bytes: Array[Byte]) =
+    val result = new StringBuilder
+    for (aByte <- bytes) {
+      result.append(String.format("%02x", aByte))
+    }
+    result.toString