make style (#11442)

2021-04-26 13:50:34 +02:00
parent 04ab2ca639
commit 32dbb2d954
105 changed files with 202 additions and 202 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -1122,7 +1122,7 @@ class ModelTesterMixin:

        # a candidate for testing_utils
        def get_current_gpu_memory_use():
-            """ returns a list of cuda memory allocations per GPU in MBs"""
+            """returns a list of cuda memory allocations per GPU in MBs"""

            per_device_memory = []
            for id in range(torch.cuda.device_count()):