Skip to content

Commit 94419bf

Browse files
committed
fixed error path for new tf version
1 parent 8ada254 commit 94419bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_serial_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,5 +84,5 @@ def graph_save_fail():
8484

8585
def test_ensure_graph_fail():
8686
"""Test that super_serial fails in graph mode."""
87-
with pytest.raises(tf.python.framework.errors_impl.NotFoundError):
87+
with pytest.raises(tf.errors.NotFoundError):
8888
graph_save_fail()

0 commit comments

Comments
 (0)