Skip to content

Commit 969dde8

Browse files
authored
Merge branch 'main' into FSTORE-1564-new
2 parents afe70f4 + 3b53fff commit 969dde8

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

python/hsfs/engine/python.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1473,9 +1473,9 @@ def _start_offline_materialization(offline_write_options: Dict[str, Any]) -> boo
14731473
def _convert_feature_log_to_df(feature_log, cols) -> pd.DataFrame:
14741474
if feature_log is None and cols:
14751475
return pd.DataFrame(columns=cols)
1476-
if not (isinstance(feature_log, (list, pd.DataFrame, pl.DataFrame))) or (
1476+
if not (isinstance(feature_log, (list, pd.DataFrame, pl.DataFrame)) or (
14771477
HAS_NUMPY and isinstance(feature_log, np.ndarray)
1478-
):
1478+
)):
14791479
raise ValueError(f"Type '{type(feature_log)}' not accepted")
14801480
if isinstance(feature_log, list) or (
14811481
HAS_NUMPY and isinstance(feature_log, np.ndarray)

python/hsfs/storage_connector.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -373,12 +373,16 @@ def prepare_spark(self, path: Optional[str] = None) -> Optional[str]:
373373

374374
def connector_options(self) -> Dict[str, Any]:
375375
"""Return options to be passed to an external S3 connector library"""
376-
return {
376+
self.refetch()
377+
options = {
377378
"access_key": self.access_key,
378379
"secret_key": self.secret_key,
379380
"session_token": self.session_token,
380381
"region": self.region,
381382
}
383+
if self.arguments.get("fs.s3a.endpoint"):
384+
options["endpoint"] = self.arguments.get("fs.s3a.endpoint")
385+
return options
382386

383387
def read(
384388
self,

0 commit comments

Comments
 (0)