ModuFlow/mask-ddpm
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
MingzheYang
2026-01-23 23:57:06 +08:00
parent 0caa80b6ef
commit f10b1b5e2c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/export_samples.py
View File

@@ -257,8 +257,8 @@ def main():
row["__cond_file_id"] = str(int(cond[b].item())) if cond is not None else "-1" row["__cond_file_id"] = str(int(cond[b].item())) if cond is not None else "-1"
if args.include_time and time_col in header: if args.include_time and time_col in header:
row[time_col] = str(row_index) row[time_col] = str(row_index)
for i, c in enumerate(cont_cols): for i, c in enumerate(cont_cols):
val = float(x_cont[b, t, i]) val = float(x_cont[b, t, i])
if int_like.get(c, False): if int_like.get(c, False):
row[c] = str(int(round(val))) row[c] = str(int(round(val)))
else: else: