exposing type checking (#1022)

* exposing types under dsl.types

sdk/python/kfp/components/_components.py

@@ -26,7 +26,7 @@
 from ._structures import ComponentSpec
 from ._structures import *
 from kfp.dsl import PipelineParam
-from kfp.dsl._types import InconsistentTypeException, check_types
+from kfp.dsl.types import InconsistentTypeException, check_types
 import kfp
 
 _default_component_name = 'Component'

sdk/python/kfp/dsl/__init__.py

@@ -17,5 +17,5 @@
 from ._pipeline import Pipeline, pipeline, get_pipeline_conf
 from ._container_op import ContainerOp
 from ._ops_group import OpsGroup, ExitHandler, Condition
-from ._component import python_component
+from ._component import python_component, component
 #TODO: expose the component decorator when ready

sdk/python/kfp/dsl/_component.py

@@ -14,7 +14,7 @@
 
 from ._metadata import ComponentMeta, ParameterMeta, TypeMeta, _annotation_to_typemeta
 from ._pipeline_param import PipelineParam
-from ._types import check_types, InconsistentTypeException
+from .types import check_types, InconsistentTypeException
 import kfp
 
 def python_component(name, description=None, base_image=None, target_component_file: str = None):

sdk/python/kfp/dsl/_metadata.py

@@ -14,7 +14,7 @@
 
 from typing import Dict, List
 from abc import ABCMeta, abstractmethod
-from ._types import BaseType, _check_valid_type_dict, _instance_to_dict
+from .types import BaseType, _check_valid_type_dict, _instance_to_dict
 
 class BaseMeta(object):
   __metaclass__ = ABCMeta

sdk/python/tests/compiler/compiler_tests.py

@@ -25,7 +25,7 @@
 
 from kfp.dsl._component import component
 from kfp.dsl import ContainerOp, pipeline
-from kfp.dsl._types import Integer, InconsistentTypeException
+from kfp.dsl.types import Integer, InconsistentTypeException
 
 class TestCompiler(unittest.TestCase):
 

sdk/python/tests/components/test_components.py

@@ -22,7 +22,7 @@
 import kfp
 import kfp.components as comp
 from kfp.components._yaml_utils import load_yaml
-from kfp.dsl._types import InconsistentTypeException
+from kfp.dsl.types import InconsistentTypeException
 
 class LoadComponentTestCase(unittest.TestCase):
     def _test_load_component_from_file(self, component_path: str):

sdk/python/tests/dsl/component_tests.py

@@ -13,9 +13,9 @@
 # limitations under the License.
 
 import kfp
-from kfp.dsl._component import component
+from kfp.dsl import component
 from kfp.dsl._metadata import ComponentMeta, ParameterMeta, TypeMeta
-from kfp.dsl._types import GCSPath, Integer, InconsistentTypeException
+from kfp.dsl.types import Integer, GCSPath, InconsistentTypeException
 from kfp.dsl import ContainerOp, Pipeline
 import unittest
 

sdk/python/tests/dsl/pipeline_tests.py

@@ -15,7 +15,7 @@
 import kfp
 from kfp.dsl import Pipeline, PipelineParam, ContainerOp, pipeline
 from kfp.dsl._metadata import PipelineMeta, ParameterMeta, TypeMeta
-from kfp.dsl._types import GCSPath, Integer
+from kfp.dsl.types import GCSPath, Integer
 import unittest
 
 

sdk/python/tests/dsl/type_tests.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
-from kfp.dsl._types import _instance_to_dict, check_types, GCSPath
+from kfp.dsl.types import _instance_to_dict, check_types, GCSPath
 import unittest
 
 class TestTypes(unittest.TestCase):