clean up all __init__.py

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

Reviewed By: ajinkya-deogade

Differential Revision: D46875151

fbshipit-source-id: e19d9ac79c0a4ad1b1ab49112e36f80c55062ea4

d2go/data/dataset_mappers/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
-
-# @fb-only: from d2go.data.dataset_mappers import fb  # isort:skip  # noqa 
-from d2go.data.dataset_mappers.build import (  # noqa
+from d2go.data.dataset_mappers.build import (
     build_dataset_mapper,
     D2GO_DATA_MAPPER_REGISTRY,
 )
-from d2go.data.dataset_mappers.d2go_dataset_mapper import D2GoDatasetMapper  # noqa
-from d2go.data.dataset_mappers.rotated_dataset_mapper import (  # noqa
-    RotatedDatasetMapper,
-)
+from d2go.data.dataset_mappers.d2go_dataset_mapper import D2GoDatasetMapper
+from d2go.data.dataset_mappers.rotated_dataset_mapper import RotatedDatasetMapper
+
+__all__ = [
+    "build_dataset_mapper",
+    "D2GO_DATA_MAPPER_REGISTRY",
+    "D2GoDatasetMapper",
+    "RotatedDatasetMapper",
+]
+
+
+# Populating registreis
+# @fb-only: from d2go.data.dataset_mappers import fb as _fb  # isort:skip  # noqa 

d2go/data/transforms/__init__.py

@@ -2,14 +2,14 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 
-# import all modules to make sure Registry works
-# @fb-only: from d2go.data.transforms import fb  # isort:skip  # noqa 
+# Populating registreis
 from d2go.data.transforms import (  # noqa
-    affine,
+    affine as _affine,
     auto_aug,
-    blur,
-    box_utils,
-    color_yuv,
-    crop,
-    d2_native,
+    blur as _blur,
+    box_utils as _box_utils,
+    color_yuv as _color_yuv,
+    crop as _crop,
+    d2_native as _d2_native,
 )
+# @fb-only: from d2go.data.transforms import fb as _fb  # isort:skip  # noqa 

d2go/evaluation/__init__.py

@@ -2,7 +2,11 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 
-# @fb-only: from d2go.evaluation import fb  # noqa 
-from d2go.evaluation.prediction_count_evaluation import PredictionCountEvaluator  # noqa
+from d2go.evaluation.prediction_count_evaluation import PredictionCountEvaluator
 
-__all__ = [k for k in globals().keys() if not k.startswith("_")]
+__all__ = [
+    "PredictionCountEvaluator",
+]
+
+# Populating registreis
+# @fb-only: from d2go.evaluation import fb as _fb  # noqa 

d2go/export/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
-# enable registry
-from d2go.export import torchscript  # noqa
+# Populating registreis
+from d2go.export import torchscript as _torchscript  # noqa

d2go/modeling/__init__.py

@@ -2,5 +2,9 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 
-# NOTE: making necessary imports to register with Registery
-from d2go.modeling import backbone, meta_arch, modeldef  # noqa  # noqa  # noqa
+# Populating registreis
+from d2go.modeling import (  # noqa
+    backbone as _backbone,
+    meta_arch as _meta_arch,
+    modeldef as _modeldef,
+)

d2go/modeling/backbone/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
-# @fb-only: from d2go.modeling.backbone import fb, fbnet_v2  # noqa 
-
-
-# Explicitly expose all registry-based modules
-__all__ = [
-    "fbnet_v2",
-    # @fb-only: "fb", 
-]
+# Populating registreis
+from d2go.modeling.backbone import fbnet_v2 as _fbnet_v2  # noqa
+# @fb-only: from d2go.modeling.backbone import fb as _fb  # isort:skip  # noqa 

d2go/modeling/meta_arch/__init__.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
-# NOTE: making necessary imports to register with Registry
-# @fb-only: from d2go.modeling.meta_arch import fb  # isort:skip  # noqa 
+# Populating registreis
 from d2go.modeling.meta_arch import (  # noqa
-    fcos,
-    panoptic_fpn,
-    rcnn,
-    retinanet,
-    semantic_seg,
+    fcos as _fcos,
+    panoptic_fpn as _panoptic_fpn,
+    rcnn as _rcnn,
+    retinanet as _retinanet,
+    semantic_seg as _semantic_seg,
 )
+# @fb-only: from d2go.modeling.meta_arch import fb as _fb  # isort:skip  # noqa 

d2go/modeling/modeldef/__init__.py

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

d2go/utils/__init__.py

@@ -2,7 +2,6 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 
 
-# import to make sure Registry works
-# usort does not respect the `fb-only` tag so use isort:skip here
-# @fb-only: from d2go.utils import fb  # isort:skip  # noqa 
-from d2go.utils import flop_calculator  # noqa
+# Populating registreis
+from d2go.utils import flop_calculator as _flop_calculator  # noqa
+# @fb-only: from d2go.utils import fb as _fb  # isort:skip  # noqa