fix type calculation

Author: alamb

datafusion/core/src/physical_plan/windows/mod.rs

@@ -198,11 +198,11 @@ fn create_udwf_window_expr(
     name: String,
 ) -> Result<Arc<dyn BuiltInWindowFunctionExpr>> {
     // need to get the types into an owned vec for some reason
-    let input_types: Vec<_> = input_schema
-        .fields()
+    let input_types: Vec<_> = args
         .iter()
-        .map(|f| f.data_type().clone())
-        .collect();
+        .map(|arg| arg.data_type(input_schema).map(|dt| dt.clone()))
+        .collect::<Result<_>>()?;
+
     // figure out the output type
     let data_type = (fun.return_type)(&input_types)?;
     Ok(Arc::new(WindowUDFExpr {