Add __all__ exports to Python SDK (#587)

feast-dev · Mar 29, 2020 · 314369d · 314369d
1 parent 880269b
commit 314369d
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/sdk/python/feast/__init__.py b/sdk/python/feast/__init__.py
@@ -0,0 +1,24 @@
+from pkg_resources import DistributionNotFound, get_distribution
+
+from .client import Client
+from .entity import Entity
+from .feature import Feature
+from .feature_set import FeatureSet
+from .source import KafkaSource, Source
+from .value_type import ValueType
+
+try:
+    __version__ = get_distribution(__name__).version
+except DistributionNotFound:
+    # package is not installed
+    pass
+
+__all__ = [
+    "Client",
+    "Entity",
+    "Feature",
+    "FeatureSet",
+    "Source",
+    "KafkaSource",
+    "ValueType",
+]