Unexpected type checking with pyright

[DanielSank]

Using pyright in neovim, I see unexpected behavior in type checking:

import tunits as tu

x = (tu.ElectricPotentialArray(np.array([1, 2, 3]) * tu.V), 5 * tu.W)
reveal_type(x)  # Type of "x" is "tuple[ElectricPotentialArray, Power]"
y = x[tu.V]
reveal_type(y)  # Type of "y" is "ElectricPotentialArray | Power"

The third line should be flagged as a type error. From the fourth line, it looks like tu.V is being treated as an integer. Is it possible that the type annotations for this library have some bug?

[NoureldinYosri]

thanks for catching this. I think these overloads confuse it but I don't know why it seems to go for third overload over the first, any ideas?

wait indexing a tuple with a tu.V is undefined and it will follow the type annotation of tuple ... tu.V is not treated as integer, it is treated as a generic numeric because it inherets it

TypedUnits/tunits/core/__init__.pyi

Line 224 in efc0bb4

class Value(Generic[NumericalT], WithUnit, np.generic, SupportsIndex):