test python 311 support

mage-ai · Feb 28, 2024 · 5aca89e · 5aca89e
1 parent dd4df0f
commit 5aca89e
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 4 deletions.
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
@@ -50,7 +50,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
     needs: check-code-quality
     steps:
       - uses: actions/checkout@v3

diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.10-bookworm
+FROM python:3.11-bookworm
 LABEL description="Deploy Mage on ECS"
 ARG FEATURE_BRANCH
 USER root

diff --git a/mage_ai/data_preparation/models/constants.py b/mage_ai/data_preparation/models/constants.py
@@ -62,6 +62,9 @@ class BlockType(str, Enum):
     SENSOR = 'sensor'
     TRANSFORMER = 'transformer'
 
+    def __str__(self):
+        return self.value
+
 
 class BlockColor(str, Enum):
     BLUE = 'blue'

diff --git a/mage_integrations/requirements.txt b/mage_integrations/requirements.txt
@@ -2,7 +2,7 @@ attrs==17.4.0
 azure-storage-blob==12.14.1
 backoff
 clickhouse_sqlalchemy
-couchbase==4.1.1
+couchbase==4.1.6
 deltalake==0.7.0
 elasticsearch==8.9.0
 facebook_business==17.0.2

diff --git a/requirements.txt b/requirements.txt
@@ -83,7 +83,8 @@ redshift-connector==2.0.915
 lxml==4.9.4
 snowflake-connector-python==3.4.0
 sshtunnel==0.4.0
-tables==3.7.0
+tables==3.7.0; python_version == '3.8'
+tables==3.9.2; python_version >= '3.9'
 trino~=0.326
 
 # Azure