add top module exports

src/datachain/__init__.py

@@ -0,0 +1,34 @@
+from datachain.lib.dc import C, DataChain
+from datachain.lib.feature import Feature
+from datachain.lib.feature_utils import pydantic_to_feature
+from datachain.lib.file import File, FileError, FileFeature, IndexedFile, TarVFile
+from datachain.lib.image import ImageFile, convert_images
+from datachain.lib.text import convert_text
+from datachain.lib.udf import Aggregator, Generator, Mapper
+from datachain.lib.utils import AbstractUDF, DataChainError
+from datachain.query.dataset import UDF as BaseUDF  # noqa: N811
+from datachain.query.schema import Column
+from datachain.query.session import Session
+
+__all__ = [
+    "AbstractUDF",
+    "Aggregator",
+    "BaseUDF",
+    "C",
+    "Column",
+    "DataChain",
+    "DataChainError",
+    "Feature",
+    "File",
+    "FileError",
+    "FileFeature",
+    "Generator",
+    "ImageFile",
+    "IndexedFile",
+    "Mapper",
+    "Session",
+    "TarVFile",
+    "convert_images",
+    "convert_text",
+    "pydantic_to_feature",
+]

tests/unit/test_module_exports.py

@@ -0,0 +1,31 @@
+# flake8: noqa: F401
+
+import pytest
+
+
+def test_module_exports():
+    try:
+        from datachain import (
+            AbstractUDF,
+            Aggregator,
+            BaseUDF,
+            C,
+            Column,
+            DataChain,
+            DataChainError,
+            Feature,
+            File,
+            FileError,
+            FileFeature,
+            Generator,
+            ImageFile,
+            IndexedFile,
+            Mapper,
+            Session,
+            TarVFile,
+            convert_images,
+            convert_text,
+            pydantic_to_feature,
+        )
+    except Exception as e:  # noqa: BLE001
+        pytest.fail(f"Importing raised an exception: {e}")