private void removeState(String partitionFlowId, String groupByFlowId) {
        Map<String, State> groupByStates = states.get(partitionFlowId);
        if (groupByStates != null) {
            groupByStates.remove(groupByFlowId);
            if (groupByStates.isEmpty()) {
                states.remove(partitionFlowId);
            }
        }
    }
State holder for  partitioned use case
modules/siddhi-core/src/main/java/io/siddhi/core/util/snapshot/state/PartitionStateHolder.java