Skip to content

Commit e5d9269

Browse files
committed
fixed the integeration test cases
1 parent f3528bd commit e5d9269

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

ads/feature_store/common/spark_session_singleton.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from ads.common.decorator.runtime_dependency import OptionalDependency
1111
import os
1212
from ads.common.oci_client import OCIClientFactory
13-
from ads.feature_store.common.utils.utility import get_env_bool
1413

1514
try:
1615
from delta import configure_spark_with_delta_pip
@@ -87,6 +86,7 @@ def __init__(self, metastore_id: str = None):
8786
)
8887

8988
if not developer_enabled() and metastore_id:
89+
print("Developer not Enabled")
9090
# Get the authentication credentials for the OCI data catalog service
9191
auth = copy.copy(ads.auth.default_signer())
9292

@@ -105,6 +105,7 @@ def __init__(self, metastore_id: str = None):
105105
)
106106

107107
if developer_enabled():
108+
print("developer_enabled")
108109
# Configure spark session with delta jars only in developer mode. In other cases,
109110
# jars should be part of the conda pack
110111
self.spark_session = configure_spark_with_delta_pip(

ads/feature_store/execution_strategy/spark/spark_execution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ def _save_offline_dataframe(
292292

293293
output_details = {
294294
"error_details": error_details,
295-
"validation_output": str(validation_output),
295+
"validation_output": str(validation_output) if validation_output else None,
296296
"commit_id": "commit_id",
297297
"feature_statistics": feature_statistics,
298298
}
@@ -427,7 +427,7 @@ def _save_dataset_input(self, dataset, dataset_job: DatasetJob):
427427

428428
output_details = {
429429
"error_details": error_details,
430-
"validation_output": str(validation_output),
430+
"validation_output": str(validation_output) if validation_output else None,
431431
"commit_id": "commit_id",
432432
"feature_statistics": feature_statistics,
433433
}

0 commit comments

Comments
 (0)