Skip to content

GitLab

Commit 6f91155b authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Use pure PY3 syntax 

PiperOrigin-RevId: 326731205
parent bd9c6c4e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
official/core/base_task.py
View file @ 6f91155b
......@@ -19,14 +19,12 @@ import functools
from typing import Any, Callable, Optional
from absl import logging
import six
import tensorflow as tf
from official.modeling.hyperparams import config_definitions as cfg
@six.add_metaclass(abc.ABCMeta)
class Task(tf.Module):
class Task(tf.Module, metaclass=abc.ABCMeta):
"""A single-replica view of training procedure.
Tasks provide artifacts for training/evalution procedures, including
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment