updates

modeling/official/core/actions.py

@@ -19,12 +19,12 @@ from typing import List
 from absl import logging
 
 import gin
-import orbit
+import modeling.orbit
 import tensorflow as tf, tf_keras
 
-from official.core import base_trainer
-from official.core import config_definitions
-from official.modeling import optimization
+from modeling.official.core import base_trainer
+from modeling.official.core import config_definitions
+from modeling.official.modeling import optimization
 
 
 class PruningAction:

modeling/official/core/base_trainer.py

@@ -22,7 +22,7 @@ import functools
 from typing import Union, Optional
 from absl import logging
 import gin
-import orbit
+import modeling.orbit
 import tensorflow as tf, tf_keras
 
 from official.core import base_task

modeling/official/core/train_lib.py

@@ -21,14 +21,14 @@ from typing import Any, List, Mapping, Optional, Tuple
 # Import libraries
 
 from absl import logging
-import orbit
+import modeling.orbit
 import tensorflow as tf, tf_keras
 
-from official.core import actions
-from official.core import base_task
-from official.core import base_trainer
-from official.core import config_definitions
-from official.core import train_utils
+from modeling.official.core import actions
+from modeling.official.core import base_task
+from modeling.official.core import base_trainer
+from modeling.official.core import config_definitions
+from modeling.official.core import train_utils
 
 maybe_create_best_ckpt_exporter = train_utils.maybe_create_best_ckpt_exporter
 

modeling/official/core/train_utils.py

@@ -24,18 +24,18 @@ from typing import Any, Callable, Dict, List, Optional, Union
 from absl import logging
 import gin
 import numpy as np
-import orbit
+import modeling.orbit
 import tensorflow as tf, tf_keras
 
 # pylint: disable=g-direct-tensorflow-import
 from tensorflow.python.framework import ops
 from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2_as_graph
 # pylint: enable=g-direct-tensorflow-import
-from official.core import base_task
-from official.core import base_trainer
-from official.core import config_definitions
-from official.core import exp_factory
-from official.modeling import hyperparams
+from modeling.official.core import base_task
+from modeling.official.core import base_trainer
+from modeling.official.core import config_definitions
+from modeling.official.core import exp_factory
+from modeling.official.modeling import hyperparams
 
 
 BEST_CHECKPOINT_NAME = 'best_ckpt'