lib/models/point.rb in db_clustering-0.1.6 vs lib/models/point.rb in db_clustering-0.1.7

@@ -2,17 +2,22 @@
   module Models
     class Point
 
       attr_accessor :cluster, :is_noise, :datasource_point
 
-      def initialize(datasource_point)
+      def initialize(datasource_point:, vector_params: nil)
         @is_noise = false
         @cluster = nil
         @datasource_point = datasource_point
+        @vector_params = vector_params
       end
 
       def vector
-        vector_object = @datasource_point.clustering_vector
+        if @vector_params
+          vector_object = @datasource_point.clustering_vector(@vector_params)
+        else
+          vector_object = @datasource_point.clustering_vector
+        end
 
         if vector_object.is_a?(Hash) || vector_object.is_a?(Array)
           DbClustering::Models::Vector.new(object: vector_object)
         else
           raise "clustering_vector method needs to result to a Hash or an Array object"