According to mypy_primer, this change doesn't affect type check results on a corpus of open source code. ✅

Merging, but comments are very welcome!

This change seems to have introduced a small regression.

from enum import StrEnum

class Capability(StrEnum):
    COLOR_CONTROL = "colorControl"
    COLOR_TEMPERATURE = "colorTemperature"

def func(capability: Capability | str):
    if capability in (Capability.COLOR_CONTROL, Capability.COLOR_TEMPERATURE):
        reveal_type(capability)

# before
Capability

# now
Capability | str

It reverses part of the improvement from #20602.

Thanks for the post, but this is actually an improvement!

from enum import StrEnum

class Capability(StrEnum):
    COLOR_CONTROL = "colorControl"
    COLOR_TEMPERATURE = "colorTemperature"

def func(capability: Capability | str):
    if capability in (Capability.COLOR_CONTROL, Capability.COLOR_TEMPERATURE):
        print(capability, type(capability))

func(Capability.COLOR_CONTROL)
func("colorControl")

colorControl <enum 'Capability'>
colorControl <class 'str'>

I have tests for this fix in my stack, but it looks like I split the tests for this change into a different commit

Ideally we could narrow to a literal, but in general mypy is missing a bunch of logic around the values of enums

#20672 includes more tests here

Thanks for the post, but this is actually an improvement!

Good point, missed that! Thanks for working on it.