diff --git a/sdk/python/feast/infra/gcp.py b/sdk/python/feast/infra/gcp.py index a75aeb5ddbb..4d7711e5b9f 100644 --- a/sdk/python/feast/infra/gcp.py +++ b/sdk/python/feast/infra/gcp.py @@ -110,7 +110,7 @@ def materialize_single_feature_view( start_date=start_date, end_date=end_date, ) - table = offline_job.to_table() + table = offline_job.to_arrow() if feature_view.input.field_mapping is not None: table = _run_field_mapping(table, feature_view.input.field_mapping) diff --git a/sdk/python/feast/infra/local.py b/sdk/python/feast/infra/local.py index d04fe8d7407..0033b28a9d3 100644 --- a/sdk/python/feast/infra/local.py +++ b/sdk/python/feast/infra/local.py @@ -109,7 +109,7 @@ def materialize_single_feature_view( end_date=end_date, config=config, ) - table = offline_job.to_table() + table = offline_job.to_arrow() if feature_view.input.field_mapping is not None: table = _run_field_mapping(table, feature_view.input.field_mapping) diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index 0c961a50483..3b73248ee33 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -275,7 +275,7 @@ def _block_until_done(): print(f"Done writing to '{job_config.destination}'.") return str(job_config.destination) - def to_table(self) -> pyarrow.Table: + def to_arrow(self) -> pyarrow.Table: return self.client.query(self.query).to_arrow() diff --git a/sdk/python/feast/infra/offline_stores/file.py b/sdk/python/feast/infra/offline_stores/file.py index c61162f81f2..9ab71ec2d8c 100644 --- a/sdk/python/feast/infra/offline_stores/file.py +++ b/sdk/python/feast/infra/offline_stores/file.py @@ -38,7 +38,7 @@ def to_df(self): df = self.evaluation_function() return df - def to_table(self): + def to_arrow(self): # Only execute the evaluation function to build the final historical retrieval dataframe at the last moment. df = self.evaluation_function() return pyarrow.Table.from_pandas(df) diff --git a/sdk/python/feast/infra/offline_stores/offline_store.py b/sdk/python/feast/infra/offline_stores/offline_store.py index 6e2394b44b7..f2ee65d80de 100644 --- a/sdk/python/feast/infra/offline_stores/offline_store.py +++ b/sdk/python/feast/infra/offline_stores/offline_store.py @@ -33,7 +33,7 @@ def to_df(self) -> pd.DataFrame: pass @abstractmethod - def to_table(self) -> pyarrow.Table: + def to_arrow(self) -> pyarrow.Table: """Return dataset as pyarrow Table synchronously""" pass diff --git a/sdk/python/tensorflow_metadata/proto/v0/path_pb2.py b/sdk/python/tensorflow_metadata/proto/v0/path_pb2.py index 4b6dec828cf..d732119ead5 100644 --- a/sdk/python/tensorflow_metadata/proto/v0/path_pb2.py +++ b/sdk/python/tensorflow_metadata/proto/v0/path_pb2.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: tensorflow_metadata/proto/v0/path.proto -"""Generated protocol buffer code.""" + from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from google.protobuf import reflection as _reflection diff --git a/sdk/python/tensorflow_metadata/proto/v0/schema_pb2.py b/sdk/python/tensorflow_metadata/proto/v0/schema_pb2.py index d3bfc50616c..78fda8003da 100644 --- a/sdk/python/tensorflow_metadata/proto/v0/schema_pb2.py +++ b/sdk/python/tensorflow_metadata/proto/v0/schema_pb2.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: tensorflow_metadata/proto/v0/schema.proto -"""Generated protocol buffer code.""" + from google.protobuf.internal import enum_type_wrapper from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message diff --git a/sdk/python/tensorflow_metadata/proto/v0/statistics_pb2.py b/sdk/python/tensorflow_metadata/proto/v0/statistics_pb2.py index 21473adc75c..d8e12bd1209 100644 --- a/sdk/python/tensorflow_metadata/proto/v0/statistics_pb2.py +++ b/sdk/python/tensorflow_metadata/proto/v0/statistics_pb2.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: tensorflow_metadata/proto/v0/statistics.proto -"""Generated protocol buffer code.""" + from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from google.protobuf import reflection as _reflection diff --git a/sdk/python/tests/test_historical_retrieval.py b/sdk/python/tests/test_historical_retrieval.py index 72cf67b1785..e3c28ef8665 100644 --- a/sdk/python/tests/test_historical_retrieval.py +++ b/sdk/python/tests/test_historical_retrieval.py @@ -461,7 +461,7 @@ def test_historical_features_from_bigquery_sources( check_dtype=False, ) - table_from_sql_entities = job_from_sql.to_table() + table_from_sql_entities = job_from_sql.to_arrow() assert_frame_equal( actual_df_from_sql_entities, table_from_sql_entities.to_pandas() ) @@ -546,7 +546,7 @@ def test_historical_features_from_bigquery_sources( .reset_index(drop=True), check_dtype=False, ) - table_from_df_entities = job_from_df.to_table() + table_from_df_entities = job_from_df.to_arrow() assert_frame_equal( actual_df_from_df_entities, table_from_df_entities.to_pandas() )