message: Fix tests broken from cl/286925612.
diff: @@ -398,7 +398,10 @@ class Network(base_layer.Layer):
  @property
  def _layer_checkpoint_dependencies(self):
  """Dictionary of layer dependencies to be included in the checkpoint."""
  - if not self._is_graph_network and base_layer_utils.is_subclassed(self):
  + # Use getattr becuase this function can be called from __setattr__, at which
  + # point the _is_graph_network attribute has not been created.
  + if (not getattr(self, '_is_graph_network', False) and
  + base_layer_utils.is_subclassed(self)):
  return {} # Only add layer dependencies for graph networks
  weight_layer_index = 0
  