diff --git a/serving/src/main/java/feast/serving/service/CoreService.java b/serving/src/main/java/feast/serving/service/CoreService.java index 62a0c7b3c4..6ab9a103ef 100644 --- a/serving/src/main/java/feast/serving/service/CoreService.java +++ b/serving/src/main/java/feast/serving/service/CoreService.java @@ -142,7 +142,8 @@ public Map getAllFeatureSpecs() { */ public boolean isConnected() { ConnectivityState state = channel.getState(true); - return state.compareTo(ConnectivityState.READY) == 0; + return state == ConnectivityState.IDLE + || state == ConnectivityState.READY; } /**