diff --git a/core/src/main/scala/ONNXBytesDataSource.scala b/core/src/main/scala/ONNXBytesDataSource.scala
index def5a944..83815213 100644
--- a/core/src/main/scala/ONNXBytesDataSource.scala
+++ b/core/src/main/scala/ONNXBytesDataSource.scala
@@ -8,6 +8,7 @@ class ONNXBytesDataSource(onnxBytes: Array[Byte]) extends AutoCloseable with Dat
 
   val onnxHelper = new ONNXHelper(onnxBytes)
 
+  //TODO: return non-tensor params
   override def getParams[T: Numeric: ClassTag](name: String): Tensor[T] = {
     val params = onnxHelper.params.filter(x => x._1 == name).headOption
     params match {