Add missing type constraints for Set and ImmSet generation

Set and ImmSet elements must be arraykeys; this was previously enforced by the runtime, but not by the typechecker. In nightly builds, this is now enforced by the typechecker.