Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] authored and mtsokol committed Mar 13, 2024
1 parent 2927fff commit 4c309c7
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 9 deletions.
14 changes: 7 additions & 7 deletions sparse/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,24 +4,24 @@

from ._version import __version__, __version_tuple__ # noqa: F401


__array_api_version__ = "2022.12"


class BackendType(Enum):
pydata = "PyData"
finch = "Finch"
PyData = "PyData"
Finch = "Finch"


_ENV_VAR_NAME = "SPARSE_BACKEND"

backend_var = ContextVar("backend", default=BackendType.pydata)
backend_var = ContextVar("backend", default=BackendType.PyData)

if _ENV_VAR_NAME in os.environ:
backend_var.set(BackendType[os.environ[_ENV_VAR_NAME]])


class Backend:
def __init__(self, backend=BackendType.pydata):
def __init__(self, backend=BackendType.PyData):
self.backend = backend
self.token = None

Expand All @@ -36,9 +36,9 @@ def __exit__(self, exc_type, exc_value, traceback):
@staticmethod
def get_backend_module():
backend = backend_var.get()
if backend == BackendType.pydata:
if backend == BackendType.PyData:
import sparse.pydata_backend as backend_module
elif backend == BackendType.finch:
elif backend == BackendType.Finch:
import sparse.finch_backend as backend_module
else:
raise ValueError(f"Invalid backend identifier: {backend}")
Expand Down
2 changes: 1 addition & 1 deletion sparse/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import pytest


@pytest.fixture(scope="session", params=[sparse.BackendType.pydata, sparse.BackendType.finch])
@pytest.fixture(scope="session", params=[sparse.BackendType.PyData, sparse.BackendType.Finch])
def backend(request):
with sparse.Backend(backend=request.param):
yield request.param
2 changes: 1 addition & 1 deletion sparse/tests/test_backends.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@


def test_backend_contex_manager(backend):
if backend == sparse.BackendType.finch:
if backend == sparse.BackendType.Finch:
with pytest.raises(NotImplementedError):
sparse.COO.from_numpy(np.eye(5))
else:
Expand Down

0 comments on commit 4c309c7

Please sign in to comment.