Improved examples [skip ci]

ankane · ankane · commit b95d4a8e8ef0 · 2023-05-28T12:59:26.000-07:00
diff --git a/examples/pytorch_image_search.py b/examples/pytorch_image_search.py
@@ -10,8 +10,7 @@
 
 
 # establish connection
-conn = psycopg.connect(dbname='pgvector_example')
-conn.autocommit = True
+conn = psycopg.connect(dbname='pgvector_example', autocommit=True)
 conn.execute('CREATE EXTENSION IF NOT EXISTS vector')
 register_vector(conn)
 
@@ -38,7 +37,7 @@ def generate_embeddings(inputs):
 # generate, save, and index embeddings
 if seed:
     conn.execute('DROP TABLE IF EXISTS image')
-    conn.execute('CREATE TABLE image (id bigserial primary key, embedding vector(512))')
+    conn.execute('CREATE TABLE image (id bigserial PRIMARY KEY, embedding vector(512))')
 
     for data in tqdm(dataloader):
         embeddings = generate_embeddings(data[0])
diff --git a/tests/test_psycopg.py b/tests/test_psycopg.py
@@ -3,8 +3,7 @@
 import psycopg
 import pytest
 
-conn = psycopg.connect(dbname='pgvector_python_test')
-conn.autocommit = True
+conn = psycopg.connect(dbname='pgvector_python_test', autocommit=True)
 
 conn.execute('CREATE EXTENSION IF NOT EXISTS vector')
 conn.execute('DROP TABLE IF EXISTS item')
