Allow importing of new Python classes (feast-dev#1422)

woop · web-flow · commit aec54359a1f5 · 2021-03-28T16:41:49.000-07:00
* Expose 0.10 Python classes

Signed-off-by: Willem Pienaar &lt;git@willem.co&gt;

* Reformat imports

Signed-off-by: Willem Pienaar &lt;git@willem.co&gt;

* Fix broken import

Signed-off-by: Willem Pienaar &lt;git@willem.co&gt;

* Fix broken type

Signed-off-by: Willem Pienaar &lt;git@willem.co&gt;
diff --git a/sdk/python/feast/__init__.py b/sdk/python/feast/__init__.py
@@ -10,8 +10,10 @@
 )
 from .entity import Entity
 from .feature import Feature
+from .feature_store import FeatureStore
 from .feature_table import FeatureTable
 from .feature_view import FeatureView
+from .repo_config import RepoConfig
 from .value_type import ValueType
 
 try:
@@ -28,8 +30,10 @@
     "KafkaSource",
     "KinesisSource",
     "Feature",
+    "FeatureStore",
     "FeatureTable",
     "FeatureView",
+    "RepoConfig",
     "SourceType",
     "ValueType",
 ]
diff --git a/sdk/python/feast/infra/provider.py b/sdk/python/feast/infra/provider.py
@@ -2,7 +2,8 @@
 from datetime import datetime
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
-from feast import FeatureTable, FeatureView
+from feast.feature_table import FeatureTable
+from feast.feature_view import FeatureView
 from feast.protos.feast.types.EntityKey_pb2 import EntityKey as EntityKeyProto
 from feast.protos.feast.types.Value_pb2 import Value as ValueProto
 from feast.repo_config import RepoConfig
diff --git a/sdk/python/tests/test_client.py b/sdk/python/tests/test_client.py
@@ -983,7 +983,9 @@ def test_no_auth_sent_when_auth_disabled(
         client.list_feature_tables()
 
 
-def _ingest_test_getfeaturetable_mocked_resp(file_url: str, date_partition_col: str):
+def _ingest_test_getfeaturetable_mocked_resp(
+    file_url: str, date_partition_col: str = ""
+):
     return GetFeatureTableResponse(
         table=FeatureTableProto(
             spec=FeatureTableSpecProto(
@@ -1005,9 +1007,7 @@ def _ingest_test_getfeaturetable_mocked_resp(file_url: str, date_partition_col:
                     ),
                     event_timestamp_column="datetime",
                     created_timestamp_column="timestamp",
-                    date_partition_column=date_partition_col
-                    if date_partition_col is not None
-                    else None,
+                    date_partition_column=date_partition_col,
                 ),
             ),
             meta=FeatureTableMetaProto(),
