jobs/transforms/date_diff_job.rb in remi-0.2.37 vs jobs/transforms/date_diff_job.rb in remi-0.2.38

@@ -12,10 +12,13 @@
   define_target :target_data, Remi::DataTarget::DataFrame
 
   define_transform :main, sources: :source_data, targets: :target_data do
     Remi::SourceToTargetMap.apply(source_data.df, target_data.df) do
       map source(:date1, :date2) .target(:difference)
-        .transform(->(d1,d2) { [Date.strptime(d1), Date.strptime(d2)] })
+        .transform(->(row) {
+          row[:date1] = Date.strptime(row[:date1])
+          row[:date2] = Date.strptime(row[:date2])
+        })
         .transform(Remi::Transform::DateDiff.new(params[:measure]))
     end
   end
 end