Commit Graph

26 Commits

Author SHA1 Message Date
Christian Geier 3a67f20d4e
chore(sample): fix pipeline with secrets examples (#10245) 2023-12-01 00:30:03 +00:00
Chen Sun a727783abd
chore: remove 'dummy_root' in test samples (#9640)
* remove 'dummy_root' in test samples

* more cleanup
2023-06-21 23:51:41 +00:00
Lingqing Gan b8caf811c2
feat(backend): support secrets (#9178)
* support secret in driver

* add sample tests

* fix comments
2023-04-20 00:33:39 +00:00
Lingqing Gan ca63f20049
fix(backend): fix volume bug (#9150)
* fix volume bug

* fix indentation

* update driver image

* revert launcher image sha
2023-04-18 23:53:54 +00:00
Lingqing Gan 89bdac9340
feat(backend): support optional and default parameters. Fixes #8716 (#8765)
* support optional and default parameters

* add sample test

* change assert value

* update test

* update test

* fix sample test

* fix sample test

* fix sample test

* test fix

* change parameter logic

* test

* test

* change logic handling not provided parameter

* remove unnecessary code and add more sample case

* fix sample test bug

* fix sample test

* update logic

* change parameter logic

* update logic

* update logic handling when parameter not provided

* fix sample test

* add error message

* fix samples test

* samples test

* remove unused code
2023-03-02 17:47:04 +00:00
Lingqing Gan faca1e5aab
feat(backend): add support for environment variables. Fixes #8704 (#8886)
* support env var and add test

* sample test fix

* remove compiler test case

* fix test script

* update sample test
2023-02-27 18:02:08 +00:00
Connor McCarthy e371fa3795
feat(sdk): add compilation logic for google artifact types (support custom artifact types pt. 2) (#8232)
* remove dependency of deprecated on v2

* use new artifact identification logic

* add custom artifact type compiler test

* compile schema_version from non-kfp artifacts

* update tests

* add importer to compiler test

* update golden snapshots

* fix type checking code

* undo executor_test diff

* remove gcpc types compiler test

* add compilation test comment

* update google artifact handling

* update compiler test

* only permit system. and google. artifact types; add tests

* update compiler test

* simplify type checking logic

* use bundled artifact types in PipelineArtifactChannel

* raise error on unknown v1 input or output

* add type utils and move functions

* add type utils tests

* remove schema_version from inputspec and outputspec

* update artifact proto creation and tests

* propogate bundled artifact type string changes throughout sdk

* update golden snapshot

* remove unused v1 compat and tests

* update type checking and tests to validate schema_version using semantic versioning

* update importer sample

* update test data config

* update test_data config

* clean up sample

* clean up sample
2022-09-12 14:00:02 -06:00
Scott_Xu 121ddcbc04
fix(sdk): fix v2 sample tests for cuj3 (#8168)
* fix v2 sample tests for cuj3

* separate sh -c

* fix two step pipeline

* fix piping error

* fix nit

* change path to uri to correct cat command

* correct bash script

* remove display name

* change directory reference in commands

* remove old files
2022-08-25 15:30:22 -07:00
Scott_Xu 04c827fc3a
feat(sdk): supporting bring your own container for arbitrary input and outputs (#8066)
* support container_component decorator for function with no inputs

* resolve review comments

* add sample tests for milestone 1

* modify compiler test data

* resolve reviews

* resolve reviews

* WIP

* implementation of function of no inputs

* fixed sample test

* re-fix sample test

* fix rebase merge conflict

* resolve formatting

* resolve isort error for test data

* resolve comments

* fix nit

* resolve nit

* add implementation for placeholders i/o, sample and compiler tests

* resolve comments and merge logic for constructing container component

* resolve comments

* resolve comments

* fix assertion messages

* add error handling for accessing artifact by itself

* add test for raising error for accessing artifact by itself
2022-08-08 21:00:53 +00:00
Yaqi Ji 8bb0d0e025
feat(sdk)!: move v2 to main namespace (#7376)
* feat(sdk)!: move v2 to main namespace

* edit samples
2022-03-04 21:56:51 +00:00
Yaqi Ji b693909a08
feat(sdk)!: move v1 to deprecated folder (#7291)
* chore(sdk): move v1 to deprecated

* fix testsg

* fix testsg

* fix setup.py

* fix test

* s'

* fix tests

* fix test

* retore v2 test changes

* fix py3.6 test

* fix py3.6 test

* fix py3.6 import fallback error

* remove deprecated

* fix samples test

* sample test

* fix samples

* add readme

* restroe test

* python require

* remove path

* fix tests

* inteegration tests

* remove tfx tests for dependency with kfp v1

* fix e2e

* fix e2e

* fix integration tests

* fix sampe

* move client down

* change to kfp

* add import alias

* fix

* runid

* fix dsl

* only use kfp for function

* revert train_until_good

* tfx test

* kfp

* try import

* onprem
2022-03-04 00:52:59 +00:00
Yuan (Bob) Gong c7448c71df
test(v2): refactor sample test utils into a py package (#7105)
* test(v2): refactor sample test utils into a py package

* fix error
2021-12-23 11:10:01 +08:00
Yuan (Bob) Gong 048218b091
feat(v2): control flow - conditional (#7088)
* feat(v2): control flow - conditional

* fix v2 go tests

* fix importer sample

added condition_test

* address comments

* fix
2021-12-22 10:13:21 +08:00
Yuan (Bob) Gong f099b93a2b
test(samples): simplify v2 sample test verify funcs + clean up (#6964)
* test(samples): simplify v2 sample test verify funcs + clean up

* chore(samples): add placeholder_if tests back, fix flakiness in xgboost sample

* test(samples): clean up empty tests in config.yaml

* chore: add placeholder_if back

* chore(samples): clean up all samples

* fix

* Update xgboost_sample.py
2021-11-30 10:55:30 +00:00
Chen Sun 73804f8928
test(sample): reenable samples tests run with V2_ENGINE mode (#6932)
* reenable v2 engine

* fix

* fix

* fix

* log v2 engine test

* don't fill in default value at test util

* fix

* pass to job instead of spec
2021-11-20 18:18:35 +00:00
capri-xiyue 951b968cb6
feat(v2): Support v2 caching (#6569)
* modified argo workflow

* v2 caching draft

* move driver folder to component folder

* move driver back to driver folder

* finished cache logic, need debug

* added cached execution entry

* fixed typo

* added cache v2 e2e test

* added enable caching flag for kfp v2 tests

* fixed v2 e2e test

* fixed cache e2e test

* fixed cache e2e test

* pass container spec as a whole

* fixed nits

* fixed ut

* remove local file wriiting for v2 cache

* fixed bug
2021-09-24 17:12:40 -07:00
Chen Sun ed3453f794
chore: update comments in a pipeline sample (#6601)
* Update lightweight_python_functions_v2_pipeline.py

* Update lightweight_python_functions_v2_pipeline.py

* Update lightweight_python_functions_v2_pipeline.py

* Update mock_lightweight_python_functions_v2_pipeline.json

* Update lightweight_python_functions_v2_pipeline.json

* Update mock_lightweight_python_functions_v2_pipeline.json
2021-09-23 11:29:36 -07:00
Chen Sun 236f67ec8d
fix(samples): Add `kfp_package_path` to importer sample test. (#6507) 2021-09-03 15:49:10 -07:00
capri-xiyue 4438929259
feat(v2): added importer support (#6415)
* added importer support in compilor

* added importer launcher

* complete importer launcher draft

* added importer e2e test draft

* added importer e2e test draft and fixed build error

* added importer e2e test draft and fixed build error

* changed sample test config

* fixed mlmd publish error

* fixed format and added task verification for importer e2e test

* changed sample v2 importer test

* added non default bucket support

* fixed importer build error

* fixed e2e test

* modify importer e2e test

* fixed e2e test

* fixed e2e test

* fixed e2e test

* fixed importer e2e test

* fixed comments

* fixed unit test

* fixed importer e2e test

* added todo comment

* remove unused code

* Update importer.go

Co-authored-by: Yuan (Bob) Gong <4957653+Bobgy@users.noreply.github.com>
2021-08-31 04:06:03 -07:00
Ajay Gopinathan f3f383c2ff
chore(sdk): Refactor and move all v2 related code to under the v2 namespace. (#6358)
* Refactor and move all v2 related code to under the v2 namespace.

Most of the changes are around imports and restructuring of the
codebase. While it looks like a lot of code was added, most of the code
already existed and was simply moved or copied over to v2. The only
exceptions are:
- under kfp/v2/components/component_factory.py: some helper functions
  were copied with simplification from _python_op.py
- we no longer strip the `_path` suffix in v2 components.

Note: there is still some duplication of code (particularly between
component_factory.py and _python_op.py), but it's ok for now since we
intend to replace some of this with v2 ComponentSpec + BaseComponent.

* Update setup.py.

* update tests.

* revert accidental change of gcpc

* Fix component entrypoint.

* Update goldens.

* fix tests.

* fix merge conflict.

* revert gcpc change.

* fix tests.

* fix tests.

* Add type aliases for moved files.

* merge and update goldens.
2021-08-17 19:25:37 -07:00
Yuan (Bob) Gong 25958081e6
chore(v2): parameter passing. Fixes #6151 (#6208)
* chore(v2): publish output parameters

* chore(v2): resolve input parameters from producer task

* chore(v2): add implicit parameter dependencies to DAG tasks

* test(v2): add verification for v2 tests

* fix compiler unit test
2021-08-05 18:25:50 -07:00
Yuan (Bob) Gong 9c69e68652
test(v2): v2 sample test supports v2 engine mode (#6193)
* test(v2): v2 sample test supports v2 engine mode

* enable other tests

* chore(v2): prepublish pod info to MLMD before user command

* fix test failure

* update annotations

* fix go tests

* add use-new-run arg to kaniko

* test(v2): add resource request/limit to build image tasks

* verify

* test2

* pin 1.3.0 kaniko image to avoid OOM issues

* simplify the change

* fix

* clean up unnecessary changes
2021-08-01 01:58:37 -07:00
Yuan (Bob) Gong 9f73f91aaf
chore(samples): simplify and update v2 samples (#6076)
* chore(samples): simplify and update hello world v2 sample

* update comment

* add copyright
2021-07-19 19:35:15 -07:00
James Wu 27415fde4b
feat(backend): add V2 lightweight Python pipeline sample. Fixes #5689 (#5998) 2021-07-11 01:42:25 -07:00
Alexey Volkov cc83e1089b
Assigned copyright to the project authors (#5587) 2021-05-05 13:53:22 +08:00
Yuan (Bob) Gong d024935339
WIP: KFP v2 POC (#5092)
* Changes:
* add kfp v2 hello world sample
* generate go proto code
* code to unmarshal via jsonpb
* generate container spec
* add root DAG task and Makefile to run in argo
* env setup to quickly build driver container
* generate mlmd grpc golang client
* mlmd grpc client usage example
* driver-dag: put execution and context
* driver dag: put context only when execution is a DAG
* driver task: resolve parameter from parent execution
* chain dag driver with task driver
* driver: output argo parameters
* driver: build driver docker image
* driver: push image to dev registry
* compiler: root dag driver compiler
* mlmd doc
* driver: use task spec and executor spec as driver input instead
* driver: pod spec patch output for type EXECUTOR
* compiler: passing - parameter placeholder with hello world example
* include generated argo workflow in source control
* driver no longer outputs parameters
* publisher: publish parameters to execution custom properties
* driver, publisher: execution state
* publisher: remove built binary from repo
* sample: add producer consumer sample
* sample: producer consumer sample with only parameters
* e2e output parameter support
* e2e: driver resolves input parameter from tasks in the same DAG
* compiler: convert KFP task dependency to argo task dependency
* feat: refactor publisher so it can run as an entrypoint
* build: entrypoint image
* feat: executor and publisher in one container via entrypoint rewriting

* fixed compile error (#1)

* add licenses

* update readme

Co-authored-by: capri-xiyue <52932582+capri-xiyue@users.noreply.github.com>
2021-02-08 10:18:33 -08:00