fix(sdk.v2): Fix missing experimental modules in setup.py (#6817)
* fix missing experimental modules * fix typo * fix test
This commit is contained in:
parent
fca4c1c2fd
commit
7768f28de3
|
|
@ -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):
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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',
|
||||||
],
|
],
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue