Skip to content

Commit 6443307

Browse files
authored
Fix paiio read full table bug (#2849)
* Remove hdfs params in Python * fomat code * Cleanup submitter code * fix paiio read bug
1 parent 1d041d8 commit 6443307

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

python/runtime/pai/submitter_test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,6 @@ def test_submit_pai_train_task(self):
162162
feature_metas=iris_feature_metas,
163163
label_meta=iris_label_meta,
164164
validation_metrics="Accuracy".split(","),
165-
save="model_save",
166165
batch_size=1,
167166
epoch=1,
168167
validation_steps=1,

python/runtime/xgboost/explain.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,10 @@ def xgb_shap_dataset(datasource,
5151
# (TODO: lhw) we may specify pai_explain_table in datasoure
5252
# and discard the condition statement here
5353
conn = PaiIOConnection.from_table(pai_explain_table)
54+
stream = db.db_generator(conn, None, label_meta)
5455
else:
5556
conn = db.connect_with_data_source(datasource)
56-
stream = db.db_generator(conn, select, label_meta)
57+
stream = db.db_generator(conn, select, label_meta)
5758
selected_cols = db.selected_cols(conn, select)
5859

5960
if transform_fn:

0 commit comments

Comments
 (0)