bazel - Machine learning model test -
i have java unit test use machine learning model file, call model.bin, being built simple text file external tool (call model-builder, built through c++ rule). while make sense define genrule, doesn't support deps (model.bin & model-builder in case). see 2 options:
- make test code build
model.binfile. conceptually, not want test test (i want test reading prebuild model). - implement custom extension rule executes
./model-builder --in my-input.txt --outmodel.bin`.
what best-practise in bazel?
from description believe genrule best route. though genrule have deps sounds model.bin generated file , make sense out. model-builder looks tool generate file , make sense genrule tool.
something build file should work.
genrule( name = "model-bin", srcs = ["//path/to:my-input.txt"], outs = ["model.bin"], tools = ["//path/to:model-builder"], cmd = "$(location //path/to:model-builder) --in $< --out $@" )
Comments
Post a Comment