From b442264dc935afc1b768e9c3abcc448695de2e77 Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Wed, 27 Jul 2022 10:54:38 -0700 Subject: [PATCH] formatting fix for #1962 --- tests/unit/test_sparse_attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_sparse_attention.py b/tests/unit/test_sparse_attention.py index 4c35251f..8ff843c9 100755 --- a/tests/unit/test_sparse_attention.py +++ b/tests/unit/test_sparse_attention.py @@ -106,7 +106,7 @@ def test_bslongformersparsityconfig_module_availability(): def test_localwindowsparsityconfig_module_availability(): return True try: - from deepspeed.ops.sparse_attention import LocalSlidingWindowSparsityConfig + from deepspeed.ops.sparse_attention import LocalSlidingWindowSparsityConfig # noqa: F401 except ImportError: print("LocalSlidingWindowSparsityConfig Module is not installed!") return False -- GitLab