Is this a typo in recommenders/docs/examples/basic_retrieval.ipynb?

In class ModelensModel and NoBaseClassMovielensModel, the line of self.task in the init is:

self.task: tf.keras.layers.Layer = task

But in the quickstart.ipynb, the self.task in class Modelens is:

task: tfrs.tasks.Retrieval
self.task = task

I'm wondering if the task class belongs to the keras.layers.Layer class. Is this a typo?

Full code in quickstart:

class MovieLensModel(tfrs.Model):
  # We derive from a custom base class to help reduce boilerplate. Under the hood,
  # these are still plain Keras Models.

  def __init__(
      self,
      user_model: tf.keras.Model,
      movie_model: tf.keras.Model,
      task: tfrs.tasks.Retrieval):
    super().__init__()

    # Set up user and movie representations.
    self.user_model = user_model
    self.movie_model = movie_model

    # Set up a retrieval task.
    self.task = task

Full code in basic_retrieval:

class MovielensModel(tfrs.Model):

  def __init__(self, user_model, movie_model):
    super().__init__()
    self.movie_model: tf.keras.Model = movie_model
    self.user_model: tf.keras.Model = user_model
    self.task: tf.keras.layers.Layer = task

class NoBaseClassMovielensModel(tf.keras.Model):

  def __init__(self, user_model, movie_model):
    super().__init__()
    self.movie_model: tf.keras.Model = movie_model
    self.user_model: tf.keras.Model = user_model
    self.task: tf.keras.layers.Layer = task

Open recommenders/docs/examples/basic_retrieval.ipynb and compare the MovielensModel and NoBaseClassMovielensModel definitions with the quickstart.ipynb example shown in the issue. Check whether the task argument is defined and whether its annotation matches the retrieval task usage; done means the example is internally consistent and runs without this discrepancy.