fix(sdk.v2): Fix missing experimental modules in setup.py (#6817)

* fix missing experimental modules

* fix typo

* fix test
This commit is contained in:
Chen Sun 2021-10-27 10:58:39 -07:00 committed by GitHub
parent fca4c1c2fd
commit 7768f28de3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 9 additions and 10 deletions

View File

@ -19,10 +19,9 @@ import shutil
import tempfile import tempfile
import unittest import unittest
from kfp.v2 import components import kfp.v2.components.experimental as components
from kfp.v2.compiler.experimental import compiler from kfp.v2.compiler.experimental import compiler
import kfp.v2.dsl.experimental as dsl import kfp.v2.dsl.experimental as dsl
from kfp.v2.components.types import type_utils
class CompilerTest(unittest.TestCase): class CompilerTest(unittest.TestCase):

View File

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
"""Pipeline using ExitHandler.""" """Pipeline using ExitHandler."""
from kfp.v2 import components import kfp.v2.components.experimental as components
import kfp.v2.dsl.experimental as dsl import kfp.v2.dsl.experimental as dsl
from kfp.v2.compiler.experimental import compiler from kfp.v2.compiler.experimental import compiler

View File

@ -14,7 +14,7 @@
from typing import List from typing import List
from kfp.v2 import components import kfp.v2.components.experimental as components
import kfp.v2.dsl.experimental as dsl import kfp.v2.dsl.experimental as dsl
from kfp.v2.compiler.experimental import compiler from kfp.v2.compiler.experimental import compiler

View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from kfp.v2 import components import kfp.v2.components.experimental as components
import kfp.v2.dsl.experimental as dsl import kfp.v2.dsl.experimental as dsl
from kfp.v2.compiler.experimental import compiler from kfp.v2.compiler.experimental import compiler

View File

@ -12,9 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import pathlib import kfp.v2.components.experimental as components
from kfp.v2 import components
import kfp.v2.dsl.experimental as dsl import kfp.v2.dsl.experimental as dsl
from kfp.v2.compiler.experimental import compiler from kfp.v2.compiler.experimental import compiler

View File

@ -11,5 +11,3 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from kfp.v2.components.experimental.yaml_component import load_component_from_text

View File

@ -11,3 +11,5 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from kfp.v2.components.experimental.yaml_component import load_component_from_text

View File

@ -116,11 +116,13 @@ setup(
'kfp.notebook', 'kfp.notebook',
'kfp.v2', 'kfp.v2',
'kfp.v2.compiler', 'kfp.v2.compiler',
'kfp.v2.compiler.experimental',
'kfp.v2.components', 'kfp.v2.components',
'kfp.v2.components.types', 'kfp.v2.components.types',
'kfp.v2.components.types.experimental', 'kfp.v2.components.types.experimental',
'kfp.v2.components.experimental', 'kfp.v2.components.experimental',
'kfp.v2.dsl', 'kfp.v2.dsl',
'kfp.v2.dsl.experimental',
'kfp.v2.google.client', 'kfp.v2.google.client',
'kfp.v2.google.experimental', 'kfp.v2.google.experimental',
], ],