Have inferArgumentTypes return the InferenceContext