updates

modeling/orbit/__init__.py

@@ -14,17 +14,17 @@
 
 """Defines exported symbols for the `orbit` package."""
 
-from orbit import actions
+from modeling.orbit import actions
 # Internal import orbit.
-from orbit import utils
+from modeling.orbit import utils
 
-from orbit.controller import Action
-from orbit.controller import Controller
+from modeling.orbit.controller import Action
+from modeling.orbit.controller import Controller
 
-from orbit.runner import AbstractEvaluator
-from orbit.runner import AbstractTrainer
+from modeling.orbit.runner import AbstractEvaluator
+from modeling.orbit.runner import AbstractTrainer
 
-from orbit.standard_runner import StandardEvaluator
-from orbit.standard_runner import StandardEvaluatorOptions
-from orbit.standard_runner import StandardTrainer
-from orbit.standard_runner import StandardTrainerOptions
+from modeling.orbit.standard_runner import StandardEvaluator
+from modeling.orbit.standard_runner import StandardEvaluatorOptions
+from modeling.orbit.standard_runner import StandardTrainer
+from modeling.orbit.standard_runner import StandardTrainerOptions

modeling/orbit/actions/__init__.py

@@ -65,12 +65,12 @@ to avoid unnecessary writes by setting the `write_value` parameter to `False`
 for most clients.
 """
 
-from orbit.actions.conditional_action import ConditionalAction
+from modeling.orbit.actions.conditional_action import ConditionalAction
 
-from orbit.actions.export_saved_model import ExportFileManager
-from orbit.actions.export_saved_model import ExportSavedModel
+from modeling.orbit.actions.export_saved_model import ExportFileManager
+from modeling.orbit.actions.export_saved_model import ExportSavedModel
 
-from orbit.actions.new_best_metric import JSONPersistedValue
-from orbit.actions.new_best_metric import NewBestMetric
+from modeling.orbit.actions.new_best_metric import JSONPersistedValue
+from modeling.orbit.actions.new_best_metric import NewBestMetric
 
-from orbit.actions.save_checkpoint_if_preempted import SaveCheckpointIfPreempted
+from modeling.orbit.actions.save_checkpoint_if_preempted import SaveCheckpointIfPreempted

modeling/orbit/actions/conditional_action.py

@@ -16,8 +16,8 @@
 
 from typing import Any, Callable, Sequence, Union
 
-from orbit import controller
-from orbit import runner
+from modeling.orbit import controller
+from modeling.orbit import runner
 
 import tensorflow as tf, tf_keras
 

modeling/orbit/actions/new_best_metric.py

@@ -20,8 +20,8 @@ import sys
 from typing import Any, Callable, Optional, Union
 import uuid
 
-from orbit import runner
-from orbit import utils
+from modeling.orbit import runner
+from modeling.orbit import utils
 
 import tensorflow as tf, tf_keras
 

modeling/orbit/controller.py

@@ -21,8 +21,8 @@ from typing import Callable, Iterable, Optional, Union
 
 from absl import logging
 
-from orbit import runner
-from orbit import utils
+from modeling.orbit import runner
+from modeling.orbit import utils
 
 import tensorflow as tf, tf_keras
 

modeling/orbit/standard_runner.py

@@ -39,8 +39,8 @@ from typing import Any, Optional
 
 import dataclasses
 
-from orbit import runner
-from orbit.utils import loop_fns
+from modeling.orbit import runner
+from modeling.orbit.utils import loop_fns
 
 import tensorflow as tf, tf_keras
 

modeling/orbit/utils/__init__.py

@@ -14,17 +14,17 @@
 
 """Defines exported symbols for the `orbit.utils` package."""
 
-from orbit.utils.common import create_global_step
-from orbit.utils.common import get_value
-from orbit.utils.common import make_distributed_dataset
+from modeling.orbit.utils.common import create_global_step
+from modeling.orbit.utils.common import get_value
+from modeling.orbit.utils.common import make_distributed_dataset
 
-from orbit.utils.epoch_helper import EpochHelper
+from modeling.orbit.utils.epoch_helper import EpochHelper
 
-from orbit.utils.loop_fns import create_loop_fn
-from orbit.utils.loop_fns import create_tf_while_loop_fn
-from orbit.utils.loop_fns import LoopFnWithSummaries
+from modeling.orbit.utils.loop_fns import create_loop_fn
+from modeling.orbit.utils.loop_fns import create_tf_while_loop_fn
+from modeling.orbit.utils.loop_fns import LoopFnWithSummaries
 
-from orbit.utils.summary_manager import SummaryManager
-from orbit.utils.summary_manager_interface import SummaryManagerInterface
+from modeling.orbit.utils.summary_manager import SummaryManager
+from modeling.orbit.utils.summary_manager_interface import SummaryManagerInterface
 
-from orbit.utils.tpu_summaries import OptionalSummariesFunction
+from modeling.orbit.utils.tpu_summaries import OptionalSummariesFunction

modeling/orbit/utils/loop_fns.py

@@ -15,7 +15,7 @@
 """Utilities for creating loop functions."""
 
 from absl import logging
-from orbit.utils import tpu_summaries
+from modeling.orbit.utils import tpu_summaries
 
 import tensorflow as tf, tf_keras
 

modeling/orbit/utils/summary_manager.py

@@ -16,7 +16,7 @@
 
 import os
 
-from orbit.utils.summary_manager_interface import SummaryManagerInterface
+from modeling.orbit.utils.summary_manager_interface import SummaryManagerInterface
 
 import tensorflow as tf, tf_keras