Adjust type for kv_port

Signed-off-by: Elliot Scribner <[email protected]>
feast-dev · Oct 17, 2024 · 7f911b1 · 7f911b1
1 parent c136133
commit 7f911b1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/infra/online_stores/contrib/couchbase_online_store/couchbase.py b/sdk/python/feast/infra/online_stores/contrib/couchbase_online_store/couchbase.py
@@ -37,7 +37,7 @@ class CouchbaseOnlineStoreConfig(FeastConfigBaseModel):
     user: Optional[StrictStr] = None
     password: Optional[StrictStr] = None
     bucket_name: Optional[StrictStr] = None
-    kv_port: Optional[StrictStr] = None
+    kv_port: Optional[int] = None
 
 
 class CouchbaseOnlineStore(OnlineStore):

diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/couchbase.py b/sdk/python/tests/integration/feature_repos/universal/online_store/couchbase.py
@@ -71,7 +71,7 @@ def create_online_store(self) -> Dict[str, object]:
             "user": self.username,
             "password": self.password,
             "bucket_name": self.bucket_name,
-            "kv_port": kv_port,
+            "kv_port": int(kv_port),
         }
 
     def _wait_for_server_ready(self, base_url: str, timeout: int = 120):