Expand relative imports to absolute versions

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/545

Expanding the relative imports to absolute ones helps the autodeps down the stack.

Reviewed By: tglik

Differential Revision: D45912074

fbshipit-source-id: d42c9756dde731504ee6fd0f93cf549d71157489

d2go/modeling/backbone/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
-# @fb-only: from . import fb, fbnet_v2  # noqa 
+# @fb-only: from d2go.modeling.backbone import fb, fbnet_v2  # noqa 
 
 
 # Explicitly expose all registry-based modules

d2go/modeling/backbone/fbnet_v2.py

@@ -9,6 +9,14 @@ from typing import List
 
 import torch
 import torch.nn as nn
+from d2go.modeling.backbone.modules import (
+    KeypointRCNNConvUpsamplePredictorNoUpscale,
+    KeypointRCNNIRFPredictorNoUpscale,
+    KeypointRCNNPredictor,
+    KeypointRCNNPredictorNoUpscale,
+    MaskRCNNConv1x1Predictor,
+    RPNHeadConvRegressor,
+)
 from d2go.modeling.modeldef.fbnet_modeldef_registry import FBNetV2ModelArch
 from detectron2.layers import ShapeSpec
 from detectron2.modeling import (
@@ -23,15 +31,6 @@ from detectron2.utils.logger import log_first_n
 from mobile_cv.arch.fbnet_v2 import fbnet_builder as mbuilder
 from mobile_cv.arch.utils.helper import format_dict_expanding_list_values
 
-from .modules import (
-    KeypointRCNNConvUpsamplePredictorNoUpscale,
-    KeypointRCNNIRFPredictorNoUpscale,
-    KeypointRCNNPredictor,
-    KeypointRCNNPredictorNoUpscale,
-    MaskRCNNConv1x1Predictor,
-    RPNHeadConvRegressor,
-)
-
 
 logger = logging.getLogger(__name__)
 

d2go/modeling/meta_arch/__init__.py

@@ -2,5 +2,11 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 # NOTE: making necessary imports to register with Registry
-# @fb-only: from . import fb  # isort:skip  # noqa 
-from . import fcos, panoptic_fpn, rcnn, retinanet, semantic_seg  # noqa
+# @fb-only: from d2go.modeling.meta_arch import fb  # isort:skip  # noqa 
+from d2go.modeling.meta_arch import (  # noqa
+    fcos,
+    panoptic_fpn,
+    rcnn,
+    retinanet,
+    semantic_seg,
+)

d2go/modeling/modeldef/__init__.py

@@ -6,5 +6,5 @@
 This is the centralized place to define modeldef for all projects under D2Go.
 """
 
-# @fb-only: from . import fb  # isort:skip  # noqa 
-from . import modeldef  # noqa
+# @fb-only: from d2go.modeling.modeldef import fb  # isort:skip  # noqa 
+from d2go.modeling import modeldef  # noqa

d2go/optimizer/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
-from .build import build_optimizer_mapper
+from d2go.optimizer.build import build_optimizer_mapper
 
 __all__ = ["build_optimizer_mapper"]

d2go/quantization/modeling.py

@@ -10,16 +10,18 @@ from typing import Tuple
 import detectron2.utils.comm as comm
 import torch
 from d2go.quantization import learnable_qat
+from d2go.quantization.fx import get_convert_fx_fn, get_prepare_fx_fn
+from d2go.quantization.qconfig import (
+    set_backend_and_create_qconfig,
+    smart_decode_backend,
+)
 from detectron2.checkpoint import DetectionCheckpointer
-from detectron2.engine import HookBase, SimpleTrainer
+from detectron2.engine.train_loop import HookBase, SimpleTrainer
 from detectron2.utils.file_io import PathManager
 from mobile_cv.arch.quantization.observer import update_stat as observer_update_stat
 from mobile_cv.arch.utils import fuse_utils
 from mobile_cv.common.misc.iter_utils import recursive_iterate
 
-from .fx import get_convert_fx_fn, get_prepare_fx_fn
-from .qconfig import set_backend_and_create_qconfig, smart_decode_backend
-
 TORCH_VERSION: Tuple[int, ...] = tuple(int(x) for x in torch.__version__.split(".")[:2])
 if TORCH_VERSION > (1, 10):
     from torch.ao.quantization import convert

d2go/runner/__init__.py

@@ -5,11 +5,14 @@
 import importlib
 from typing import Optional, Type, Union
 
-from .api import RunnerV2Mixin
-from .default_runner import BaseRunner, Detectron2GoRunner, GeneralizedRCNNRunner
-from .lightning_task import DefaultTask
-from .training_hooks import TRAINER_HOOKS_REGISTRY
-
+from d2go.runner.api import RunnerV2Mixin
+from d2go.runner.default_runner import (
+    BaseRunner,
+    Detectron2GoRunner,
+    GeneralizedRCNNRunner,
+)
+from d2go.runner.lightning_task import DefaultTask
+from d2go.runner.training_hooks import TRAINER_HOOKS_REGISTRY
 
 __all__ = [
     "RunnerV2Mixin",

d2go/runner/default_runner.py

@@ -54,7 +54,8 @@ from detectron2.data import (
     build_detection_train_loader as d2_build_detection_train_loader,
     MetadataCatalog,
 )
-from detectron2.engine import AMPTrainer, hooks, SimpleTrainer
+from detectron2.engine import hooks
+from detectron2.engine.train_loop import AMPTrainer, SimpleTrainer
 from detectron2.evaluation import (
     COCOEvaluator,
     DatasetEvaluators,

d2go/runner/training_hooks.py

@@ -7,7 +7,7 @@ from d2go.config import CfgNode
 
 from d2go.utils.gpu_memory_profiler import log_memory_snapshot, record_memory_history
 
-from detectron2.engine import HookBase
+from detectron2.engine.train_loop import HookBase
 from detectron2.utils.registry import Registry
 
 

d2go/setup.py

@@ -23,7 +23,8 @@ from d2go.distributed import (
     get_local_rank,
     get_num_processes_per_machine,
 )
-from d2go.runner import BaseRunner, DefaultTask, import_runner, RunnerV2Mixin
+from d2go.runner import BaseRunner, import_runner, RunnerV2Mixin
+from d2go.runner.lightning_task import DefaultTask
 from d2go.utils.helper import run_once
 from d2go.utils.launch_environment import get_launch_environment
 from d2go.utils.logging import initialize_logging, replace_print_with_logging

d2go/utils/__init__.py

@@ -4,5 +4,5 @@
 
 # import to make sure Registry works
 # usort does not respect the `fb-only` tag so use isort:skip here
-# @fb-only: from . import fb  # isort:skip  # noqa 
-from . import flop_calculator  # noqa
+# @fb-only: from d2go.utils import fb  # isort:skip  # noqa 
+from d2go.utils import flop_calculator  # noqa

d2go/utils/misc.py

@@ -12,11 +12,10 @@ from typing import Any, Callable, Dict, Iterator, Optional
 import detectron2.utils.comm as comm
 import torch
 from d2go.config import CfgNode
+from d2go.utils.tensorboard_log_util import get_tensorboard_log_dir  # noqa: forwarding
 from detectron2.utils.file_io import PathManager
 from tabulate import tabulate
 
-from .tensorboard_log_util import get_tensorboard_log_dir  # noqa: forwarding
-
 logger = logging.getLogger(__name__)
 
 # Subdirectory with model configurations dumped by the training binary.