chore: fix pipeline job schedule system test

PiperOrigin-RevId: 570128541

Author: sararob

tests/system/aiplatform/test_pipeline_job_schedule.py

@@ -34,18 +34,19 @@ class TestPipelineJobSchedule(e2e_base.TestEndToEnd):
 
     def test_create_get_pause_resume_update_list(self, shared_state):
         # Components:
-        @dsl.component
         def train(
             number_of_epochs: int,
             learning_rate: float,
         ):
             print(f"number_of_epochs={number_of_epochs}")
             print(f"learning_rate={learning_rate}")
 
+        train_op = dsl.component(train)
+
         # Pipeline:
         @dsl.pipeline(name="system-test-training-pipeline")
         def training_pipeline(number_of_epochs: int = 2):
-            train(
+            train_op(
                 number_of_epochs=number_of_epochs,
                 learning_rate=0.1,
             )