Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion aiter/jit/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,8 @@ def update_config_files(file_path: str, merge_name: str):
untuned_path = f"{AITER_ROOT_DIR}/aiter/configs/{untuned_name}.csv"
if os.path.exists(untuned_path):
untunedf = pd.read_csv(untuned_path)
keys = untunedf.columns
keys = untunedf.columns.to_list()
keys.append("cu_num")
merge_df = (
merge_df.sort_values("us")
.drop_duplicates(subset=keys, keep="first")
Expand Down
Loading