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

* fix missing experimental modules

* fix typo

* fix test

sdk/python/kfp/v2/compiler/experimental/compiler_test.py

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

sdk/python/kfp/v2/compiler_cli_tests/test_data/experimental_pipeline_with_exit_handler.py

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

sdk/python/kfp/v2/compiler_cli_tests/test_data/experimental_pipeline_with_loops.py

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

sdk/python/kfp/v2/compiler_cli_tests/test_data/experimental_pipeline_with_nested_conditions_yaml.py

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

sdk/python/kfp/v2/compiler_cli_tests/test_data/experimental_two_step_pipeline.py

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

sdk/python/kfp/v2/components/__init__.py

@@ -11,5 +11,3 @@
 # 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.
-
-from kfp.v2.components.experimental.yaml_component import load_component_from_text

sdk/python/kfp/v2/components/experimental/__init__.py

@@ -11,3 +11,5 @@
 # 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.
+
+from kfp.v2.components.experimental.yaml_component import load_component_from_text

sdk/python/setup.py

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