We really need infer_shape to not be present when we can't do it.
Remove print statements in tests.