message: Parameter Server Distribution strategy: Do not try to remove from trainable collection if not present.
diff: @@ -317,6 +317,7 @@ class ParameterServerStrategyExtended(
  if kwargs.get("trainable", True):
  collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
  l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
  + if v in l:
  l.remove(v)
  g.add_to_collections(collections, wrapped)
  elif ops.GraphKeys.GLOBAL_STEP in collections:
  