Skip to content

Commit 06bbb1b

Browse files
author
sindchad
committed
removed unnecessary print statements
1 parent 698f51c commit 06bbb1b

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/aixpert/data_construction/stage_7_final/data_final_train.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ def main() -> None:
3838
print(f"Loading → {input_path}")
3939
data = load_jsonl(input_path)
4040

41-
# Initialize buckets
4241
buckets: Dict[Tuple[int, int], List[Dict[str, Any]]] = {
4342
(0, 1): [],
4443
(1, 0): [],
@@ -58,15 +57,14 @@ def main() -> None:
5857

5958
final_rows: List[Dict[str, Any]] = []
6059

61-
# Sampling logic
6260
for key, req_count in target_counts.items():
6361
pool = buckets[key]
6462
available = len(pool)
6563

6664
print(f"\nBucket {key}: available={available}, required={req_count}")
6765

6866
if available < req_count:
69-
print("⚠️ Sampling WITH replacement.")
67+
print("Sampling WITH replacement.")
7068
sampled = random.choices(pool, k=req_count)
7169
else:
7270
sampled = random.sample(pool, req_count)

src/aixpert/data_construction/utils/dpo_transform_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,13 @@ def transform_dataset(input_path: Path, output_path: Path) -> None:
4444
print(f"Loading → {input_path}")
4545
items = load_jsonl(input_path)
4646

47-
print(f"⚙️ Transforming {len(items)} items…")
47+
print(f"Transforming {len(items)} items…")
4848
transformed = [process_item(it) for it in tqdm(items)]
4949

5050
print(f"Saving → {output_path}")
5151
write_jsonl(output_path, transformed)
5252

5353
print("\n=======================================")
54-
print("TRANSFORMATION COMPLETE")
54+
print("TRANSFORMATION COMPLETE")
5555
print(f"Total items: {len(items)}")
5656
print("=======================================\n")

0 commit comments

Comments
 (0)