Improved examples[skip ci]

Author: ankane

examples/colbert/approximate.py

@@ -6,9 +6,6 @@
 import psycopg
 import warnings
 
-# ignore warnings from colbert
-warnings.filterwarnings('ignore')
-
 conn = psycopg.connect(dbname='pgvector_example', autocommit=True)
 
 conn.execute('CREATE EXTENSION IF NOT EXISTS vector')
@@ -38,6 +35,9 @@
 $$ LANGUAGE SQL
 """)
 
+# ignore warnings from colbert
+warnings.filterwarnings('ignore')
+
 config = ColBERTConfig(doc_maxlen=220, query_maxlen=32)
 checkpoint = Checkpoint('colbert-ir/colbertv2.0', colbert_config=config, verbose=0)
 

examples/colbert/exact.py

@@ -4,9 +4,6 @@
 import psycopg
 import warnings
 
-# ignore warnings from colbert
-warnings.filterwarnings('ignore')
-
 conn = psycopg.connect(dbname='pgvector_example', autocommit=True)
 
 conn.execute('CREATE EXTENSION IF NOT EXISTS vector')
@@ -32,6 +29,9 @@
 $$ LANGUAGE SQL
 """)
 
+# ignore warnings from colbert
+warnings.filterwarnings('ignore')
+
 config = ColBERTConfig(doc_maxlen=220, query_maxlen=32)
 checkpoint = Checkpoint('colbert-ir/colbertv2.0', colbert_config=config, verbose=0)