| """Minimal smoke test for HI-Mapper (no weights).""" | |
| import torch | |
| from hi_mapper import DivHiMapper | |
| B, D = 2, 768 | |
| mapper = DivHiMapper(feat_dim=D, curvature=0.2, target_radius=1.0) | |
| root, mid, leaves, hier_loss, prompt_loss, attr_loss = mapper( | |
| torch.randn(B, 5, D), torch.randn(B, D) | |
| ) | |
| print("root", tuple(root.shape), "hier_loss", float(hier_loss)) | |