old_contents: from typing import Tuple, Callable
  
  from ..model import Model
  from ..config import registry
  from ..types import Ragged, List2d
  
  
  InT = Ragged
  OutT = List2d
  
  
  @registry.layers("ragged2list.v1")
  def ragged2list() -> Model[InT, OutT]:
      """Transform sequences from a ragged format into lists."""
      return Model("ragged2list", forward)
  
  
  def forward(model: Model[InT, OutT], Xr: InT, is_train: bool) -> Tuple[OutT, Callable]:
      lengths = Xr.lengths
  
      def backprop(dXs: OutT) -> InT:
          return Ragged(model.ops.flatten(dXs, pad=0), lengths)  # type: ignore
  
      data = model.ops.unflatten(Xr.data, Xr.lengths)  # type: ignore
      return data, backprop
  
new_contents: from typing import Tuple, Callable
  
  from ..model import Model
  from ..config import registry
  from ..types import Ragged, List2d
  
  
  InT = Ragged
  OutT = List2d
  
  
  @registry.layers("ragged2list.v1")
  def ragged2list() -> Model[InT, OutT]:
      """Transform sequences from a ragged format into lists."""
      return Model("ragged2list", forward)
  
  
  def forward(model: Model[InT, OutT], Xr: InT, is_train: bool) -> Tuple[OutT, Callable]:
      lengths = Xr.lengths
  
      def backprop(dXs: OutT) -> InT:
          return Ragged(model.ops.flatten(dXs, pad=0), lengths)  # type: ignore
  
      data = model.ops.unflatten(Xr.dataXd, Xr.lengths) # type: ignore
      return data, backprop
  
subject: Fix shape omission in ragged2list