Commit 35685bfa authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Checkpointable: Stop overriding __class__ in restore-on-create initializers

Seems to work fine without it, and the fact that it was returning a tuple was an issue (but I think is valid Python). These objects get unwrapped quickly anyway.

The odd thing is that this code has decent unit test coverage. Reproducing whatever isinstance caching in a unit test doesn't sound too helpful, since it'd basically be asserting that __class__ does not return a tuple.

PiperOrigin-RevId: 208671395
parent 87e56623
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment