def lookup_creator(next_creator, *args, **kwargs):
if load_context.in_load_context:
return (ps_values.RestoredDistributedTable(
self._container_strategy(), lambda: next_creator(*args, **kwargs))) # pylint: disable=protected-access
else:
return ps_values.DistributedTable(self._container_strategy(),
lambda: next_creator(*args, **kwargs)) # pylint: disable=protected-access
def lookup_creator(next_creator, *args, **kwargs):
return ps_values.DistributedTable(
self._container_strategy(),
lambda: next_creator(*args, **kwargs)) # pylint: disable=protected-access
