Switched to columns_hash

ankane · May 5, 2024 · fb76a12 · fb76a12
1 parent caf8798
commit fb76a12
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lib/neighbor/vector.rb b/lib/neighbor/vector.rb
@@ -39,8 +39,7 @@ def self.cast(value, dimensions:, normalize:, column_info:)
     end
 
     def self.column_info(model, attribute_name)
-      attribute_name = attribute_name.to_s
-      column = model.columns.detect { |c| c.name == attribute_name }
+      column = model.columns_hash[attribute_name.to_s]
       {
         type: column.try(:type),
         dimensions: column.try(:limit)