Improve module import (#235)

Summary:
Pull Request resolved: #235

Added `__init__.py`. Prepare for running tests with `pytest`

Reviewed By: kirklandsign

Differential Revision: D49062889

fbshipit-source-id: 71e70c19c8527dfec3fbb7f03c4e90b82d4eadee

Author: guangy10

examples/export/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from .utils import export_to_edge, export_to_exec_prog, save_pte_program
+
+__all__ = [
+    "export_to_edge",
+    "export_to_exec_prog",
+    "save_pte_program",
+]

examples/models/__init__.py

@@ -4,6 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+from .model_factory import EagerModelFactory
 
 MODEL_NAME_TO_MODEL = {
     "mul": ("toy_model", "MulModule"),
@@ -21,5 +22,7 @@
     "resnet50": ("resnet", "ResNet50Model"),
 }
 
-
-__all__ = [MODEL_NAME_TO_MODEL]
+__all__ = [
+    "EagerModelFactory",
+    "MODEL_NAME_TO_MODEL",
+]

examples/models/mobilebert/model.py

@@ -12,9 +12,6 @@
 
 from ..model_base import EagerModelBase
 
-FORMAT = "[%(filename)s:%(lineno)s] %(message)s"
-logging.basicConfig(format=FORMAT)
-
 
 class MobileBertModelExample(EagerModelBase):
     def __init__(self):