From 68e42caa846a1e65836937a99208a80cc0ac5b2f Mon Sep 17 00:00:00 2001 From: Saeed Maleki Date: Mon, 23 Oct 2023 04:25:37 +0000 Subject: [PATCH] Lint --- python/test/test_mscclpp.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/python/test/test_mscclpp.py b/python/test/test_mscclpp.py index 70c7194c4..76841f4cc 100644 --- a/python/test/test_mscclpp.py +++ b/python/test/test_mscclpp.py @@ -9,7 +9,15 @@ import netifaces as ni import pytest -from mscclpp import Fifo, Host2DeviceSemaphore, Host2HostSemaphore, ProxyService, SmDevice2DeviceSemaphore, Transport, get_ib_device_count +from mscclpp import ( + Fifo, + Host2DeviceSemaphore, + Host2HostSemaphore, + ProxyService, + SmDevice2DeviceSemaphore, + Transport, + get_ib_device_count, +) from ._cpp import _ext from .mscclpp_group import MscclppGroup from .mscclpp_mpi import MpiGroup, parametrize_mpi_groups, mpi_group @@ -19,6 +27,7 @@ skipif_ib = pytest.mark.skipif(get_ib_device_count() == 0, reason="no IB device") + def parametrize_transport(*transports: list): def decorator(func): params = [] @@ -28,6 +37,7 @@ def decorator(func): else: params.append(transport) return pytest.mark.parametrize("transport", params)(func) + return decorator