Improved type hints [skip ci]

Author: ankane

Makefile

@@ -4,7 +4,7 @@ lint:
 	pycodestyle . --ignore=E501 --exclude=.venv
 
 check:
-	ty check pgvector tests
+	ty check pgvector tests --ignore unused-type-ignore-comment
 
 build:
 	python3 -m build

pgvector/psycopg2/register.py

@@ -7,7 +7,7 @@
 
 # note: register_adapter is always global
 def register_vector(conn_or_curs: connection | cursor, globally: bool = False, arrays: bool = True) -> None:
-    conn = conn_or_curs if hasattr(conn_or_curs, 'cursor') else conn_or_curs.connection
+    conn = conn_or_curs if isinstance(conn_or_curs, connection) else conn_or_curs.connection
     cur = conn.cursor(cursor_factory=cursor)
     scope = None if globally else conn_or_curs
 

pgvector/sqlalchemy/bit.py

@@ -22,7 +22,7 @@ def bind_processor(self, dialect: Dialect) -> Any:
 
             def process(value: Any) -> Any:
                 if isinstance(value, str):
-                    return asyncpg.BitString(value)
+                    return asyncpg.BitString(value)  # type: ignore
                 return value
             return process
         else: