diff --git a/serving/src/main/java/feast/serving/service/CoreService.java b/serving/src/main/java/feast/serving/service/CoreService.java index 3c8ca076e9..85c77c271d 100644 --- a/serving/src/main/java/feast/serving/service/CoreService.java +++ b/serving/src/main/java/feast/serving/service/CoreService.java @@ -176,7 +176,8 @@ public Map getAllStorageSpecs() { */ public boolean isConnected() { ConnectivityState state = channel.getState(true); - return state.compareTo(ConnectivityState.READY) == 0; + return state == ConnectivityState.IDLE + || state == ConnectivityState.READY; } /**