Improved encode function

pgvector · Jul 9, 2024 · 954777c · 954777c
1 parent f58345e
commit 954777c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/pgvector.rb b/lib/pgvector.rb
@@ -8,7 +8,7 @@ module Pgvector
  autoload :PG, "pgvector/pg"
 
  def self.encode(data)
- if data.is_a?(SparseVector)
+ if data.is_a?(Vector) || data.is_a?(HalfVector) || data.is_a?(SparseVector)
  data.to_s
  else
  Vector.new(data).to_s