[library/python] Fix some style tests

см https://a.yandex-team.ru/review/4511498/details?checkId=29700000007055&dialogId=CiCard&filter=resultType%28RT_STYLE_CHECK%29%3BsuiteCategory%28CATEGORY_CHANGED%29&iterationType=FULL&number=1&openedItems=955241967903872625%3ART_STYLE_SUITE_CHECK%2C12143061788616128922%3ART_STYLE_SUITE_CHECK%2C8023375544570871616%3ART_STYLE_SUITE_CHECK%2C10700057700434354061%3ART_STYLE_SUITE_CHECK%2C10919096887128703016%3ART_STYLE_SUITE_CHECK%2C8720280302786645361%3ART_STYLE_SUITE_CHECK%2C16643406941049220442%3ART_STYLE_SUITE_CHECK%2C12501056625690179524%3ART_STYLE_SUITE_CHECK&snippetViewMode=word-wrap&status=STATUS_FAILED

1 files changed, 6 insertions, 3 deletions

diff --git a/library/python/pytest/plugins/ya.py b/library/python/pytest/plugins/ya.py
index def3534d76c..a02ef6852ec 100644
--- a/library/python/pytest/plugins/ya.py
+++ b/library/python/pytest/plugins/ya.py
@@ -2,7 +2,6 @@
 
 import base64
 import errno
-import re
 import sys
 import os
 import logging
@@ -15,7 +14,6 @@ import signal
 import inspect
 import warnings
 
-import attr
 import faulthandler
 import py
 import pytest
@@ -68,6 +66,7 @@ SHUTDOWN_REQUESTED = False
 
 pytest_config = None
 
+
 def configure_pdb_on_demand():
     import signal
 
@@ -185,6 +184,7 @@ def pytest_addoption(parser):
 def from_ya_test():
     return "YA_TEST_RUNNER" in os.environ
 
+
 @pytest.hookimpl(tryfirst=True)
 def pytest_configure(config):
     global pytest_config
@@ -344,8 +344,11 @@ def _collect_test_rusage(item):
         def add_metric(attr_name, metric_name=None, modifier=None):
             if not metric_name:
                 metric_name = attr_name
+
             if not modifier:
-                modifier = lambda x: x
+                def modifier(x):
+                    return x
+
             if hasattr(item.rusage, attr_name):
                 ya_inst.set_metric_value(metric_name, modifier(getattr(finish_rusage, attr_name) - getattr(item.rusage, attr_name)))