social-data-attribution-w2/src/data_attribution/attribution/influence_test.py

from pathlib import Path

import torch

from data_attribution.attribution.index import _import_attributor


def main():
    device = "cuda" if torch.cuda.is_available() else "cpu"
    Attributor = _import_attributor()

    print("before index attributor", flush=True)
    Attributor(
        Path("runs/data_small"),
        device=device,
        unit_norm=True,
    )
    print("after index attributor", flush=True)


if __name__ == "__main__":
    main()