fix issues with nanvl

Author: kumarUjjawal

datafusion/functions/src/math/nanvl.rs

@@ -107,20 +107,17 @@ impl ScalarUDFImpl for NanvlFunc {
                 .cast_to(args.return_type(), None);
         }
 
-        let target_type = if args
-            .args
-            .iter()
-            .all(|arg| matches!(arg.data_type(), Float32))
-        {
-            Float32
-        } else {
-            Float64
-        };
+        let target_type = args.return_type();
+        if !matches!(target_type, Float32 | Float64) {
+            return exec_err!(
+                "Unsupported return type {target_type:?} for function nanvl"
+            );
+        }
 
         let casted_args = args
             .args
             .iter()
-            .map(|arg| arg.cast_to(&target_type, None))
+            .map(|arg| arg.cast_to(target_type, None))
             .collect::<Result<Vec<_>>>()?;
 
         make_scalar_function(nanvl, vec![])(&casted_args)