module-import-not-at-top-of-file ignore after os.environ

[Pastells]

Sometimes you need to set an environment variable before importing a package, for example:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "4" 
os.environ["WORLD_SIZE"] = "1"
import torch

See: https://discuss.pytorch.org/t/runtimeerror-device-0-device-num-gpus-internal-assert-failed/178118/6

My suggestion is to ignore E402 module-import-not-at-top-of-file if it comes after a os.environ statement, similar to #9047

[charliermarsh]

I think I agree with this. We already ignore sys.path manipulations, and for me this is similar.

[charliermarsh]

Added in #10066, though it'll be --preview-only until we release v0.3.0, per our versioning policy.

[ion-elgreco]

@charliermarsh it might make sense to allow warnings.filterwarnings in between as well. I sometimes have to add that in between imports to silence the warnings

[Cjkjvfnby]

I just use extend-per-file-ignores in my Ruff config, to suppress this rule for that specific file.

In my case it's custom logger configuration. I need to setup logger, before calling any getLogger. Since I use logger in other files, I have to do it before imports.