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):