Commit 095bc035 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Move registry utils to core.

PiperOrigin-RevId: 330583627
parent eb34e04d
...@@ -15,8 +15,8 @@ ...@@ -15,8 +15,8 @@
# ============================================================================== # ==============================================================================
"""Experiment factory methods.""" """Experiment factory methods."""
from official.core import registry
from official.modeling.hyperparams import config_definitions as cfg from official.modeling.hyperparams import config_definitions as cfg
from official.utils import registry
_REGISTERED_CONFIGS = {} _REGISTERED_CONFIGS = {}
......
...@@ -20,7 +20,7 @@ from __future__ import division ...@@ -20,7 +20,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import tensorflow as tf import tensorflow as tf
from official.utils import registry from official.core import registry
class RegistryTest(tf.test.TestCase): class RegistryTest(tf.test.TestCase):
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
# ============================================================================== # ==============================================================================
"""A global factory to register and access all registered tasks.""" """A global factory to register and access all registered tasks."""
from official.utils import registry from official.core import registry
_REGISTERED_TASK_CLS = {} _REGISTERED_TASK_CLS = {}
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
# ============================================================================== # ==============================================================================
"""A global factory to access NLP registered data loaders.""" """A global factory to access NLP registered data loaders."""
from official.utils import registry from official.core import registry
_REGISTERED_DATA_LOADER_CLS = {} _REGISTERED_DATA_LOADER_CLS = {}
......
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