Skip to content

Commit

Permalink
Lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Saeed Maleki committed Nov 8, 2023
1 parent 842805f commit 2afce03
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion python/test/test_mscclpp.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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 = []
Expand All @@ -28,6 +37,7 @@ def decorator(func):
else:
params.append(transport)
return pytest.mark.parametrize("transport", params)(func)

return decorator


Expand Down

0 comments on commit 2afce03

Please sign in to comment.