patrick-kidger/jaxtyping

Undefined name in Ruff: F821

krzysztofrusek opened this issue · 2 comments

I the following code ruff 0.1.14 complains abut dimension annotation.

class VariationalInferenceThompsonSampling(AbstractAgent):
    mcs: Int[Array, "n"]
    tx_power: Float[Array, " n"]

For mcs I got error Undefined name n Ruff F821
It looks like adding a space at the beginning solve this, but I am not sure is this is correct.

O... thank you for pointing this.