From 7f911b11753d2aaf60a8deda0cb0dc934dd2f1ac Mon Sep 17 00:00:00 2001 From: Elliot Scribner Date: Wed, 16 Oct 2024 16:06:02 -0700 Subject: [PATCH] Adjust type for `kv_port` Signed-off-by: Elliot Scribner --- .../online_stores/contrib/couchbase_online_store/couchbase.py | 2 +- .../feature_repos/universal/online_store/couchbase.py | 2 +- 2 files changed, 2 insertions(+), 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 index 3f3e2bf2f96..91ce56a5caf 100644 --- 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 index c894740df05..f2ba12da8da 100644 --- 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):