[MASTER]
extension-pkg-whitelist=flash_attn_2_cuda,
                        torch,
                        transformer_engine_torch,
                        transformer_engine_jax

extension-pkg-allow-list=transformer_engine.transformer_engine_jax

disable=too-many-locals,
        too-few-public-methods,
        too-many-public-methods,
        too-many-positional-arguments,
        invalid-name,
        too-many-arguments,
        abstract-method,
        arguments-differ,
        too-many-instance-attributes,
        unsubscriptable-object,
        import-outside-toplevel,
        too-many-statements,
        import-error,
        too-many-lines,
        use-maxsplit-arg,
        protected-access,
        pointless-string-statement,
        cyclic-import,
        duplicate-code,
        no-member,
        attribute-defined-outside-init,
        global-statement,
        too-many-branches,
        global-variable-not-assigned,
        redefined-argument-from-local,
        line-too-long,
        too-many-return-statements,
        too-many-nested-blocks

[TYPECHECK]
ignored-modules=torch
ignored-classes=torch
