Task.__init__(params) gets a type annotation, not a default value.

PiperOrigin-RevId: 315322201

Task.init(params) gets a type annotation, not a default value.
PiperOrigin-RevId: 315322201
ec7265be · Hongkun Yu · A. Unique TensorFlower · 3269c84b · ec7265be
Commit ec7265be authored Jun 08, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Jun 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/core/base_task.py official/core/base_task.py +1 -1

No files found.
--- a/official/core/base_task.py
+++ b/official/core/base_task.py
@@ -34,7 +34,7 @@ class Task(tf.Module):
  # Special keys in train/validate step returned logs.
  loss = "loss"

-  def __init__(self, params=cfg.TaskConfig):
+  def __init__(self, params: cfg.TaskConfig):
    self._task_config = params

  @property