build: add pyarrow extra

ibis-project · Mar 27, 2023 · 2f63ce1 · 2f63ce1
1 parent fde79fa
commit 2f63ce1
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 12 deletions.
diff --git a/.github/workflows/ibis-backends.yml b/.github/workflows/ibis-backends.yml
@@ -99,6 +99,16 @@ jobs:
               - postgres
             sys-deps:
               - libgeos-dev
+          - name: postgres
+            title: PostgreSQL + PyArrow
+            extras:
+              - postgres
+              - geospatial
+              - pyarrow
+            services:
+              - postgres
+            sys-deps:
+              - libgeos-dev
           - name: impala
             title: Impala
             serial: true
@@ -188,6 +198,18 @@ jobs:
                 - postgres
               sys-deps:
                 - libgeos-dev
+          - os: windows-latest
+            backend:
+              name: postgres
+              title: PostgreSQL + PyArrow
+              extras:
+                - postgres
+                - geospatial
+                - pyarrow
+              services:
+                - postgres
+              sys-deps:
+                - libgeos-dev
           - os: windows-latest
             backend:
               name: impala

diff --git a/ci/schema/postgresql.sql b/ci/schema/postgresql.sql
@@ -53,7 +53,11 @@ CREATE TABLE awards_players (
     "yearID" BIGINT,
     "lgID" TEXT,
     tie TEXT,
-    notes TEXT
+    notes TEXT,
+    search TSVECTOR GENERATED ALWAYS AS (
+      setweight(to_tsvector('simple', notes), 'A')::TSVECTOR
+    ) STORED,
+    simvec VECTOR GENERATED always AS ('[1,2,3]'::VECTOR) STORED
 );
 
 DROP TABLE IF EXISTS functional_alltypes CASCADE;
@@ -209,13 +213,3 @@ CREATE TABLE map (kv HSTORE);
 INSERT INTO map VALUES
     ('a=>1,b=>2,c=>3'),
     ('d=>4,e=>5,c=>6');
-
-ALTER TABLE awards_players
-ADD search tsvector
-GENERATED always AS (
-  setweight(to_tsvector('simple', notes), 'A') :: tsvector
-) stored;
-
-ALTER TABLE awards_players
-ADD simvec vector
-GENERATED always AS ('[1,2,3]' :: vector) stored;
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -193,6 +193,7 @@ mysql = ["sqlalchemy", "pymysql", "sqlalchemy-views"]
 pandas = ["regex"]
 polars = ["polars", "pyarrow"]
 postgres = ["psycopg2", "sqlalchemy", "sqlalchemy-views"]
+pyarrow = ["pyarrow"]
 pyspark = ["pyarrow", "pyspark", "sqlalchemy"]
 snowflake = [
   "snowflake-connector-python",