remove prefix (#6)

data-intuitive · Nov 8, 2024 · 1aebde0 · 1aebde0
1 parent 3c44df5
commit 1aebde0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/src/dummy_anndata/generate_dataset.py b/src/dummy_anndata/generate_dataset.py
@@ -11,7 +11,7 @@
 def generate_dataset(
     n_obs=10,
     n_vars=20,
-    x_type="generate_integer_matrix",
+    x_type="integer_matrix",
     layer_types=None,
     obs_types=None,
     var_types=None,

diff --git a/src/dummy_anndata/generate_matrix.py b/src/dummy_anndata/generate_matrix.py
@@ -18,27 +18,27 @@ def int_mtx(n_obs, n_vars):
 # Possible matrix generators
 # integer matrices do not support NAs in Python
 matrix_generators = {
-    "generate_float_matrix": lambda n_obs, n_vars: float_mtx(n_obs, n_vars),
-    "generate_float_matrix_nas": lambda n_obs, n_vars: float_mtx(
+    "float_matrix": lambda n_obs, n_vars: float_mtx(n_obs, n_vars),
+    "float_matrix_nas": lambda n_obs, n_vars: float_mtx(
         n_obs, n_vars, NAs=True
     ),
-    "generate_float_csparse": lambda n_obs, n_vars: sp.sparse.csc_matrix(
+    "float_csparse": lambda n_obs, n_vars: sp.sparse.csc_matrix(
         float_mtx(n_obs, n_vars)
     ),
-    "generate_float_csparse_nas": lambda n_obs, n_vars: sp.sparse.csc_matrix(
+    "float_csparse_nas": lambda n_obs, n_vars: sp.sparse.csc_matrix(
         float_mtx(n_obs, n_vars, NAs=True)
     ),
-    "generate_float_rsparse": lambda n_obs, n_vars: sp.sparse.csr_matrix(
+    "float_rsparse": lambda n_obs, n_vars: sp.sparse.csr_matrix(
         float_mtx(n_obs, n_vars)
     ),
-    "generate_float_rsparse_nas": lambda n_obs, n_vars: sp.sparse.csr_matrix(
+    "float_rsparse_nas": lambda n_obs, n_vars: sp.sparse.csr_matrix(
         float_mtx(n_obs, n_vars, NAs=True)
     ),
-    "generate_integer_matrix": lambda n_obs, n_vars: int_mtx(n_obs, n_vars),
-    "generate_integer_csparse": lambda n_obs, n_vars: sp.sparse.csc_matrix(
+    "integer_matrix": lambda n_obs, n_vars: int_mtx(n_obs, n_vars),
+    "integer_csparse": lambda n_obs, n_vars: sp.sparse.csc_matrix(
         int_mtx(n_obs, n_vars)
     ),
-    "generate_integer_rsparse": lambda n_obs, n_vars: sp.sparse.csr_matrix(
+    "integer_rsparse": lambda n_obs, n_vars: sp.sparse.csr_matrix(
         int_mtx(n_obs, n_vars)
     ),
 }