import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
import numpy as np
from datasets import load_dataset
import sklearn_submission
from triangulation import predict_wireframe_tracks

print("Loading dataset...")
dataset = load_dataset('usm3d/hoho22k_2026_trainval', split='train', streaming=True, trust_remote_code=True)
samples = [s for idx, s in enumerate(dataset) if idx < 3]

for idx, sample in enumerate(samples):
    print(f"Sample {idx}")
    try:
        tv, te = predict_wireframe_tracks(sample)
        print(f"Tracks: {len(tv)} vertices, {len(te)} edges")
    except Exception as e:
        import traceback
        traceback.print_exc()
        print(f"CRASH: {e}")