Change: typing BigInt to Long type (#208)

Co-authored-by: Łukasz Ciołecki <[email protected]>
prinz-nussknacker · Dec 13, 2021 · fce076b · fce076b
1 parent cad48b5
commit fce076b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/...ml/src/main/scala/pl/touk/nussknacker/prinz/pmml/converter/PMMLSignatureInterpreter.scala b/...ml/src/main/scala/pl/touk/nussknacker/prinz/pmml/converter/PMMLSignatureInterpreter.scala
@@ -11,19 +11,19 @@ object PMMLSignatureInterpreter {
     case "decimal" => Typed[BigDecimal]
     case "float" => Typed[Float]
     case "double" => Typed[Double]
-    case "integer" => Typed[BigInt]
-    case "nonPositiveInteger" => Typed[BigInt]
-    case "negativeInteger" => Typed[BigInt]
+    case "integer" => Typed[Long]
+    case "nonPositiveInteger" => Typed[Long]
+    case "negativeInteger" => Typed[Long]
     case "long" => Typed[Long]
     case "int" => Typed[Int]
     case "short" => Typed[Short]
     case "byte" => Typed[Byte]
-    case "nonNegativeInteger" => Typed[BigInt]
+    case "nonNegativeInteger" => Typed[Long]
     case "unsignedLong" => Typed[Long]
     case "unsignedInt" => Typed[Int]
     case "unsignedShort" => Typed[Short]
     case "unsignedByte" => Typed[Byte]
-    case "positiveInteger" => Typed[BigInt]
+    case "positiveInteger" => Typed[Long]
     case _ => throw new IllegalArgumentException(s"Unknown PMMLDataType: $typeName")
     //TODO Potentially add support for other types from XMLSchema-2 like time and date
   }

diff --git a/prinz_pmml/src/test/scala/pl/touk/nussknacker/prinz/pmml/container/PMMLHttpApiTest.scala b/prinz_pmml/src/test/scala/pl/touk/nussknacker/prinz/pmml/container/PMMLHttpApiTest.scala
@@ -56,6 +56,6 @@ class PMMLHttpApiTest extends PMMLContainerUnitTest
   )
 
   override def expectedFraudOutputs: List[(String, typing.TypingResult)] = List(
-    ("fraud", Typed[BigInt])
+    ("fraud", Typed[Long])
   )
 }
diff --git a/...z_pmml/src/test/scala/pl/touk/nussknacker/prinz/pmml/container/PMMLLocalFileApiTest.scala b/...z_pmml/src/test/scala/pl/touk/nussknacker/prinz/pmml/container/PMMLLocalFileApiTest.scala
@@ -55,6 +55,6 @@ class PMMLLocalFileApiTest extends PMMLContainerUnitTest
   )
 
   override def expectedFraudOutputs: List[(String, typing.TypingResult)] = List(
-    ("fraud", Typed[BigInt])
+    ("fraud", Typed[Long])
   )
 }