docs: @VisibleForTesting
  	@SuppressWarnings("unchecked")
  	@Override
  	public int numKeyValueStateEntries() {
  		int sum = 0;
  		for (StateSnapshotRestore state : registeredKVStates.values()) {
  			sum += ((StateTable<?, ?, ?>) state).size();
  		}
  		return sum;
  	}