Enable TypeEvalContext#assumeType only if python.use.better.control.flow.type.inference is enabled

GitOrigin-RevId: da7e3452bfbdbfd618dbe166e5a6c199f28cae6f
This commit is contained in:
evgeny.bovykin
2025-10-15 00:30:08 +00:00
committed by intellij-monorepo-bot
parent 52ec309c76
commit 2123dbfa15
@@ -193,6 +193,9 @@ public sealed class TypeEvalContext {
@ApiStatus.Internal
public <R> @Nullable R assumeType(@NotNull PyTypedElement element, @Nullable PyType type, @NotNull Function<TypeEvalContext, R> func) {
if (!Registry.is("python.use.better.control.flow.type.inference")) {
return func.apply(this);
}
if (getKnownType(element) != null) {
// Temporary solution, as overwriting known type might introduce inconsistencies with its dependencies.
return null;