diff --git a/src/nested_pandas/datasets/__init__.py b/src/nested_pandas/datasets/__init__.py
index 4b8e827..d90b287 100644
--- a/src/nested_pandas/datasets/__init__.py
+++ b/src/nested_pandas/datasets/__init__.py
@@ -1 +1 @@
-from .generation import *  # noqa
+from .generation import generate_data, generate_parquet_file  # noqa