def _decouple_state_and_input(
        input_features: List["proto.Model_pb2.FeatureDescription"],
    ) -> Tuple[
        List["proto.Model_pb2.FeatureDescription"], List["proto.Model_pb2.FeatureDescription"]
    ]:
        """
        Utils separates state input from non-state input features.
        """
        state_features = []
        non_state_input_features = []

        for input in input_features:
            if input.type.WhichOneof("Type") == "stateType":
                state_features.append(input)
            else:
                non_state_input_features.append(input)

        return state_features, non_state_input_features

        Utils separates state input from non-state input features.
        
coremltools/converters/mil/backend/mil/load.py