|
27 | 27 | import pipeline_container_no_input
|
28 | 28 | import pipeline_with_env
|
29 | 29 | import producer_consumer_param
|
| 30 | +import subdagio |
30 | 31 | import two_step_pipeline_containerized
|
31 |
| -# import subdagio |
32 |
| - |
33 | 32 |
|
34 | 33 | _MINUTE = 60 # seconds
|
35 | 34 | _DEFAULT_TIMEOUT = 5 * _MINUTE
|
@@ -65,23 +64,16 @@ def test(self):
|
65 | 64 | # TestCase(pipeline_func=pipeline_with_volume.pipeline_with_volume),
|
66 | 65 | # TestCase(pipeline_func=pipeline_with_secret_as_volume.pipeline_secret_volume),
|
67 | 66 | # TestCase(pipeline_func=pipeline_with_secret_as_env.pipeline_secret_env),
|
68 |
| - |
69 |
| - # This next set of tests needs to be commented out until issue |
70 |
| - # https://github.com/kubeflow/pipelines/issues/11239#issuecomment-2374792592 |
71 |
| - # is addressed or the driver image that is used in CI is updated |
72 |
| - # because otherwise the tests are run against incompatible version |
73 |
| - # of the driver. In the meantime, for local validation, these tests |
74 |
| - # can be executed (once you've manually deployed an updated driver |
75 |
| - # image). |
76 |
| - |
77 |
| - # TestCase(pipeline_func=subdagio.parameter.crust), |
78 |
| - # TestCase(pipeline_func=subdagio.parameter_cache.crust), |
79 |
| - # TestCase(pipeline_func=subdagio.mixed_parameters.crust), |
80 |
| - # TestCase(pipeline_func=subdagio.multiple_parameters_namedtuple.crust), |
81 |
| - # TestCase(pipeline_func=subdagio.parameter_oneof.crust), |
82 |
| - # TestCase(pipeline_func=subdagio.artifact_cache.crust), |
83 |
| - # TestCase(pipeline_func=subdagio.artifact.crust), |
84 |
| - # TestCase(pipeline_func=subdagio.multiple_artifacts_namedtuple.crust), |
| 67 | + TestCase(pipeline_func=subdagio.parameter.crust), |
| 68 | + TestCase(pipeline_func=subdagio.parameter_cache.crust), |
| 69 | + TestCase(pipeline_func=subdagio.mixed_parameters.crust), |
| 70 | + TestCase( |
| 71 | + pipeline_func=subdagio.multiple_parameters_namedtuple.crust), |
| 72 | + TestCase(pipeline_func=subdagio.parameter_oneof.crust), |
| 73 | + TestCase(pipeline_func=subdagio.artifact_cache.crust), |
| 74 | + TestCase(pipeline_func=subdagio.artifact.crust), |
| 75 | + TestCase( |
| 76 | + pipeline_func=subdagio.multiple_artifacts_namedtuple.crust), |
85 | 77 | ]
|
86 | 78 |
|
87 | 79 | with ThreadPoolExecutor() as executor:
|
|
0 commit comments