@SuppressWarnings("unchecked")
    public T removeContext(String key) {
        if (currentState.context.containsKey(key)) {
            currentState.context.remove(key);
        } else {
            // Parent states might have a value for this context key, so explicitly set it to null in this context.
            currentState.context.put(key, null);
        }
        return (T) this;
    }
Removes a named key-value pair from the context of the current state.

<p>This method has no effect if the parent state defines the context key value pair.

@param key Key to add to remove from the current context.
@return Returns self.
smithy-utils/src/main/java/software/amazon/smithy/utils/AbstractCodeWriter.java