Add __init__.py PiperOrigin-RevId: 276533341
diff --git a/bindings/python/pyiree/BUILD b/bindings/python/pyiree/BUILD index 836dd4b..30f6b21 100644 --- a/bindings/python/pyiree/BUILD +++ b/bindings/python/pyiree/BUILD
@@ -36,11 +36,22 @@ main = "dummy.py", python_version = "PY3", deps = [ - ":binding", # build_cleaner: keep + ":pyiree", # build_cleaner: keep "//bindings/python:pathsetup", # build_cleaner: keep ], ) +py_library( + name = "pyiree", + srcs = [ + "__init__.py", + ], + srcs_version = "PY3", + deps = [ + ":binding", + ], +) + iree_py_extension( name = "binding", srcs = [
diff --git a/bindings/python/pyiree/__init__.py b/bindings/python/pyiree/__init__.py new file mode 100644 index 0000000..88528cc --- /dev/null +++ b/bindings/python/pyiree/__init__.py
@@ -0,0 +1,36 @@ +"""Module init for the python bindings.""" + +# Copyright 2019 Google LLC +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +# pylint: disable=invalid-name +# pylint: disable=g-import-not-at-top + + +def __import_submodules(): + """Force a relative sub-module import. + + Python makes this hard when the package is at the top and this works + around it, ensuring that submodules are loaded into the pyiree namespace. + """ + import importlib + + def import_rel(m): + abs_m = __package__ + "." + m + importlib.import_module(abs_m) + + import_rel("binding") + + +__import_submodules()