Fix training test (#184)

* Fix SDK training test: Add wait time before refresh

* Fix training tests in canaries

Author: zhaoqizqwang

test/integration_tests/training/cli/test_cli_training.py

@@ -27,16 +27,6 @@ def test_list_clusters(self, cluster_name):
         """Test listing clusters """
         assert cluster_name
 
-    def test_set_cluster_context(self, cluster_name):
-        """Test setting cluster context."""
-        result = execute_command([
-            "hyp", "set-cluster-context",
-            "--cluster-name", cluster_name
-        ])
-        assert result.returncode == 0
-        context_line = result.stdout.strip().splitlines()[-1]
-        assert any(text in context_line for text in ["Updated context", "Added new context"])
-
     def test_get_cluster_context(self):
         """Test getting current cluster context."""
         result = execute_command(["hyp", "get-cluster-context"])

test/integration_tests/training/sdk/test_sdk_training.py

@@ -70,10 +70,9 @@ def test_list_jobs(self, pytorch_job):
         job_names = [job.metadata.name for job in jobs]
         assert pytorch_job.metadata.name in job_names
 
-    #
     def test_refresh_job(self, pytorch_job):
         pytorch_job.refresh()
-        time.sleep(15)
+        time.sleep(30)
         assert pytorch_job.status is not None, "Job status should not be None"
         logger.info(f"Refreshed job status:\n{yaml.dump(pytorch_job.status)}")