Skip to content

Commit 324320a

Browse files
bubriksSirOibaf
authored andcommitted
[FSTORE-1864] remove reconcile_schema from client reading (#668)
1 parent df14096 commit 324320a

File tree

2 files changed

+0
-33
lines changed

2 files changed

+0
-33
lines changed

python/hsfs/engine/spark.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434
import tzlocal
3535
from hopsworks_common.core.constants import HAS_NUMPY, HAS_PANDAS
3636
from hsfs.constructor import query
37-
from hsfs.core import feature_group_api
3837

3938
# in case importing in %%local
4039
from hsfs.core.vector_db_client import VectorDbClient
@@ -223,8 +222,6 @@ def register_hudi_temporary_table(
223222
read_options,
224223
)
225224

226-
self.reconcile_schema(hudi_fg_alias, read_options, hudi_engine_instance)
227-
228225
def register_delta_temporary_table(
229226
self, delta_fg_alias, feature_store_id, feature_store_name, read_options
230227
):
@@ -241,28 +238,6 @@ def register_delta_temporary_table(
241238
read_options,
242239
)
243240

244-
self.reconcile_schema(delta_fg_alias, read_options, delta_engine_instance)
245-
246-
def reconcile_schema(self, fg_alias, read_options, engine_instance):
247-
if sorted(self._spark_session.table(fg_alias.alias).columns) != sorted(
248-
[feature.name for feature in fg_alias.feature_group._features]
249-
+ hudi_engine.HudiEngine.HUDI_SPEC_FEATURE_NAMES
250-
if fg_alias.feature_group.time_travel_format == "HUDI"
251-
else []
252-
):
253-
full_fg = feature_group_api.FeatureGroupApi().get(
254-
feature_store_id=fg_alias.feature_group._feature_store_id,
255-
name=fg_alias.feature_group.name,
256-
version=fg_alias.feature_group.version,
257-
)
258-
259-
self.update_table_schema(full_fg)
260-
261-
engine_instance.register_temporary_table(
262-
fg_alias,
263-
read_options,
264-
)
265-
266241
def _return_dataframe_type(self, dataframe, dataframe_type):
267242
if dataframe_type.lower() in ["default", "spark"]:
268243
return dataframe

python/tests/engine/test_spark.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,6 @@ def test_register_hudi_temporary_table(self, mocker):
207207
# Arrange
208208
mock_hudi_engine = mocker.patch("hsfs.core.hudi_engine.HudiEngine")
209209
mocker.patch("hsfs.feature_group.FeatureGroup.from_response_json")
210-
mock_reconcile_schema = mocker.patch(
211-
"hsfs.engine.spark.Engine.reconcile_schema"
212-
)
213210

214211
spark_engine = spark.Engine()
215212

@@ -227,15 +224,11 @@ def test_register_hudi_temporary_table(self, mocker):
227224

228225
# Assert
229226
assert mock_hudi_engine.return_value.register_temporary_table.call_count == 1
230-
assert mock_reconcile_schema.call_count == 1
231227

232228
def test_register_delta_temporary_table(self, mocker):
233229
# Arrange
234230
mock_delta_engine = mocker.patch("hsfs.core.delta_engine.DeltaEngine")
235231
mocker.patch("hsfs.feature_group.FeatureGroup.from_response_json")
236-
mock_reconcile_schema = mocker.patch(
237-
"hsfs.engine.spark.Engine.reconcile_schema"
238-
)
239232

240233
spark_engine = spark.Engine()
241234

@@ -253,7 +246,6 @@ def test_register_delta_temporary_table(self, mocker):
253246

254247
# Assert
255248
assert mock_delta_engine.return_value.register_temporary_table.call_count == 1
256-
assert mock_reconcile_schema.call_count == 1
257249

258250
def test_return_dataframe_type_default(self, mocker):
259251
# Arrange

0 commit comments

Comments
 (0)