PDL::Slatec::polyfit(): the weight ($w) dims need to match the data dims

This addresses the issue <https://sourceforge.net/p/pdl/bugs/368/>, <#48>.
PDLPorters · Mar 5, 2015 · 1df5338 · 1df5338
1 parent d245a61
commit 1df5338
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Lib/Slatec/slatec.pd b/Lib/Slatec/slatec.pd
@@ -579,6 +579,10 @@ sub PDL::polyfit {
 
   my ($x_in, $y_in, $w_in, $maxdeg_in, $eps_in) = @_;
 
+  # if $w_in does not match the data vectors ($x_in, $y_in), then we can resize
+  # it to match the size of $y_in :
+  $w_in += $y_in->zeros;
+
   # Create the output arrays
   my $r = PDL->null;