From 7fe6fc4dac2fcd6e5c212fcd8bd22a504cfc1d00 Mon Sep 17 00:00:00 2001
From: Louise Deconinck <louise.deconinck@gmail.com>
Date: Thu, 19 Sep 2024 21:29:01 +0200
Subject: [PATCH] Expose functions

---
 src/dummy_anndata/__init__.py           | 18 ++++++++++++++++++
 src/dummy_anndata/generate_dataframe.py |  2 +-
 src/dummy_anndata/generate_dataset.py   |  8 ++++----
 src/dummy_anndata/generate_dict.py      |  4 ++--
 4 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/src/dummy_anndata/__init__.py b/src/dummy_anndata/__init__.py
index 7bc9bd9..43f47fb 100644
--- a/src/dummy_anndata/__init__.py
+++ b/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")
diff --git a/src/dummy_anndata/generate_dataframe.py b/src/dummy_anndata/generate_dataframe.py
index 509053a..8fad0bb 100644
--- a/src/dummy_anndata/generate_dataframe.py
+++ b/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):
diff --git a/src/dummy_anndata/generate_dataset.py b/src/dummy_anndata/generate_dataset.py
index 0092956..7290df2 100644
--- a/src/dummy_anndata/generate_dataset.py
+++ b/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(
diff --git a/src/dummy_anndata/generate_dict.py b/src/dummy_anndata/generate_dict.py
index 85d55cd..dfb12cd 100644
--- a/src/dummy_anndata/generate_dict.py
+++ b/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