lib/remi/data_subject/data_frame.rb in remi-0.2.34 vs lib/remi/data_subject/data_frame.rb in remi-0.2.35

@@ -10,11 +10,11 @@
 
     # Public: Called to extract data from the source.
     #
     # Returns data in a format that can be used to create a dataframe.
     def extract!
-      @extract = []
+      @extract = @data.transpose
     end
 
     # Public: Converts extracted data to a dataframe
     #
     # Returns a Remi::DataFrame
@@ -22,10 +22,11 @@
       DataFrame.create(@remi_df_type, extract, order: @fields.keys)
     end
 
     private
 
-    def init_df(*args, **kargs, &block)
+    def init_df(*args, data: [], **kargs, &block)
+      @data = data
     end
   end
 
 
   class DataTarget::DataFrame < Remi::DataSubject