fix: rely on interface for literal handling

substrait-io · Feb 22, 2025 · 0394f98 · 0394f98
1 parent c7e0f02
commit 0394f98
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/isthmus/src/main/java/io/substrait/isthmus/expression/LiteralConverter.java b/isthmus/src/main/java/io/substrait/isthmus/expression/LiteralConverter.java
@@ -104,9 +104,8 @@ public Expression.Literal convert(RexLiteral literal) {
         }
         throw new UnsupportedOperationException("Unable to handle char type: " + val);
       }
-      case FLOAT, DOUBLE -> fp64(n, (Double) literal.getValue());
-      case REAL -> fp32(n, ((Double) literal.getValue()).floatValue());
-
+      case FLOAT, DOUBLE -> fp64(n, ((Number) literal.getValue()).doubleValue());
+      case REAL -> fp32(n, ((Number) literal.getValue()).floatValue());
       case DECIMAL -> {
         BigDecimal bd = bd(literal);
         yield decimal(n, bd, literal.getType().getPrecision(), literal.getType().getScale());