Debugging python tests in TensorFlow

Question

We want to debug Python tests in TensorFlow such as sparse_split_op_test and string_to_hash_bucket_op_test

The other c++ tests we could debug using gdb, however we cannot find a way to debug python tests.

Is there a way in which we can debug specific python test case run via Bazel test command (for example, bazel test //tensorflow/python/kernel_tests:sparse_split_op_test)

Allen Lavoie · Accepted Answer · 2016-11-01 01:23:42Z

3

I would first build the test:

bazel build //tensorflow/python/kernel_tests:sparse_split_op_test

Then use pdb on the resulting Python binary:

pdb bazel-bin/tensorflow/python/kernel_tests/sparse_split_op_test

That seems to work for me stepping through the first few lines of the test.

answered Nov 1, 2016 at 1:23

Allen Lavoie

5,8081 gold badge20 silver badges26 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

user3064538 Over a year ago

Don't forget to re-build the file every time you make a change.

Allen Lavoie Over a year ago

Fair point. More recently I've just been inserting import pdb; pdb.set_trace() and using bazel run, which keeps everything up to date. bazel test gets in the way of the REPL.

user3064538 Over a year ago

doesn't work if you're referencing $TEST_TMPDIR in your tests.

Collectives™ on Stack Overflow

Debugging python tests in TensorFlow

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related