python_library(
    name = "place_embedding_lib",
    srcs = [
        "config.py",
        "embed.py",
        "eval.py",
        "model.py",
        "util.py",
    ],
    py_version = "<3",
    deps = [
        "//analytics:bamboo",
        "//caffe2:torch",
        "//pytorch/hiveio:hiveio",
    ],
    external_deps = [
        ("argparse", "any"),
        ("matplotlib", "any"),
        ("numpy", "any"),
        "python-tqdm",
    ],
)

python_binary(
    name = "main",
    srcs = [
        "main.py",
    ],
    main_module = "experimental.carlyang.place_embedding.main",
    py_version = "<3",
    deps = [
        ":place_embedding_lib",
        "//visdom/py/visdom:visdom",
    ],
)

python_binary(
    name = "batch",
    srcs = [
        "batch.py",
    ],
    main_module = "experimental.carlyang.place_embedding.batch",
    py_version = "<3",
    deps = [
        ":place_embedding_lib",
        "//visdom/py/visdom:visdom",
    ],
)

python_binary(
    name = "multithread",
    srcs = [
        "multithread.py",
    ],
    main_module = "experimental.carlyang.place_embedding.multithread",
    py_version = "<3",
    deps = [
        ":place_embedding_lib",
        "//visdom/py/visdom:visdom",
    ],
)

python_binary(
    name = "baseline",
    srcs = [
        "baseline.py",
    ],
    main_module = "experimental.carlyang.place_embedding.baseline",
    py_version = "<3",
    deps = [
        ":place_embedding_lib",
        "//visdom/py/visdom:visdom",
    ],
    external_deps = [
        ("scikit-learn", "any"),
        ("numpy", "any"),
    ],
)

python_binary(
    name = "pair_gen",
    srcs = [
        "pair_gen.py",
    ],
    main_module = "experimental.carlyang.place_embedding.pair_gen",
    py_version = "<3",
    deps = [
        "//analytics:bamboo",
    ],
    external_deps = [
        ("pandas", "any"),
    ],
)
