Expose functions

src/dummy_anndata/__init__.py

@@ -1,3 +1,21 @@
 from importlib.metadata import version
 
+from .generate_vector import generate_vector, vector_generators
+from .generate_matrix import generate_matrix, matrix_generators
+from .generate_dict import generate_scalar, generate_dict, scalar_generators
+from .generate_dataframe import generate_dataframe
+from .generate_dataset import generate_dataset
+
+__all__ = [
+    "generate_vector",
+    "generate_matrix",
+    "generate_scalar",
+    "generate_dict",
+    "generate_dataframe",
+    "generate_dataset",
+    "vector_generators",
+    "matrix_generators",
+    "scalar_generators"
+]
+
 __version__ = version("dummy-anndata")

src/dummy_anndata/generate_dataframe.py

@@ -1,5 +1,5 @@
 import pandas as pd
-from generate_vector import vector_generators
+from .generate_vector import vector_generators
 
 
 def generate_dataframe(n_rows, types=None):

src/dummy_anndata/generate_dataset.py

@@ -1,9 +1,9 @@
 import anndata as ad
 
-from generate_matrix import matrix_generators
-from generate_vector import vector_generators
-from generate_dataframe import generate_dataframe
-from generate_dict import scalar_generators, generate_dict
+from .generate_matrix import matrix_generators
+from .generate_vector import vector_generators
+from .generate_dataframe import generate_dataframe
+from .generate_dict import scalar_generators, generate_dict
 
 
 def generate_dataset(

src/dummy_anndata/generate_dict.py

@@ -1,5 +1,5 @@
-from generate_vector import vector_generators
-from generate_matrix import matrix_generators
+from .generate_vector import vector_generators
+from .generate_matrix import matrix_generators
 
 import pandas as pd
 import numpy as np