API: Rename PyData backend to Numba (#665)

Author: mtsokol

asv.conf.json

@@ -15,7 +15,7 @@
     },
     {
       "env": {
-        "SPARSE_BACKEND": "PyData"
+        "SPARSE_BACKEND": "Numba"
       },
       "req": {
         "finch-tensor": ""
@@ -29,7 +29,7 @@
     "env": {
       "SPARSE_BACKEND": [
         "Finch",
-        "PyData"
+        "Numba"
       ]
     },
     "env_nobuild": {},

benchmarks/benchmark_backends.py

@@ -15,7 +15,7 @@ def setup(self):
         rng = np.random.default_rng(0)
 
         random_kwargs = {"density": 0.01, "random_state": rng}
-        if sparse.BackendType.PyData == BACKEND:
+        if sparse.BackendType.Numba == BACKEND:
             random_kwargs["format"] = "gcxs"
 
         self.s1 = sparse.random((100, 10), **random_kwargs)
@@ -51,7 +51,7 @@ def setup(self, lazy_mode, size_and_density):
         size, density = size_and_density
 
         random_kwargs = {"density": density, "random_state": rng}
-        if sparse.BackendType.PyData == BACKEND:
+        if sparse.BackendType.Numba == BACKEND:
             random_kwargs["format"] = "gcxs"
 
         self.M = sparse.random((size, size), **random_kwargs)
@@ -77,9 +77,9 @@ def fn(tns1, tns2, tns3):
                 def fn(tns1, tns2, tns3):
                     return tns1 @ tns2 + tns3
 
-        elif sparse.BackendType.PyData == BACKEND:
+        elif sparse.BackendType.Numba == BACKEND:
             if lazy_mode:
-                raise SkipNotImplemented("PyData doesn't have lazy mode")
+                raise SkipNotImplemented("Numba doesn't have lazy mode")
 
             def fn(tns1, tns2, tns3):
                 return tns1 @ tns2 + tns3
@@ -101,7 +101,7 @@ def setup(self):
         rng = np.random.default_rng(0)
 
         random_kwargs = {"density": 0.01, "random_state": rng}
-        if sparse.BackendType.PyData == BACKEND:
+        if sparse.BackendType.Numba == BACKEND:
             random_kwargs["format"] = "gcxs"
 
         self.s1 = sparse.random((100, 10), **random_kwargs)

pyproject.toml

@@ -48,7 +48,7 @@ Repository = "https://github.com/pydata/sparse.git"
 Discussions = "https://github.com/pydata/sparse/discussions"
 
 [project.entry-points.numba_extensions]
-init = "sparse.pydata_backend._numba_extension:_init_extension"
+init = "sparse.numba_backend._numba_extension:_init_extension"
 
 [tool.setuptools.packages.find]
 where = ["."]

sparse/__init__.py

@@ -8,20 +8,20 @@
 
 
 class BackendType(Enum):
-    PyData = "PyData"
+    Numba = "Numba"
     Finch = "Finch"
 
 
 _ENV_VAR_NAME = "SPARSE_BACKEND"
 
-backend_var = ContextVar("backend", default=BackendType.PyData)
+backend_var = ContextVar("backend", default=BackendType.Numba)
 
 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.Numba):
         self.backend = backend
         self.token = None
 
@@ -35,8 +35,8 @@ def __exit__(self, exc_type, exc_value, traceback):
     @staticmethod
     def get_backend_module():
         backend = backend_var.get()
-        if backend == BackendType.PyData:
-            import sparse.pydata_backend as backend_module
+        if backend == BackendType.Numba:
+            import sparse.numba_backend as backend_module
         elif backend == BackendType.Finch:
             import sparse.finch_backend as backend_module
         else:
@@ -45,8 +45,8 @@ def get_backend_module():
 
 
 def __getattr__(attr):
-    if attr == "pydata_backend":
-        import sparse.pydata_backend as backend_module
+    if attr == "numba_backend":
+        import sparse.numba_backend as backend_module
 
         return backend_module
     if attr == "finch_backend":