Increasing max test time. Also adjusting new test scripts as the don'…

…t seem to be getting used

assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py

@@ -12,7 +12,7 @@
 
 BUILD_CONTEXT = Path("../context")
 JOB_SOURCE_CODE = "../../acpt-tests/src"
-TIMEOUT_MINUTES = os.environ.get("timeout_minutes", 60)
+TIMEOUT_MINUTES = os.environ.get("timeout_minutes", 120)
 STD_LOG = Path("artifacts/user_logs/std_log.txt")
 os.environ['DS_ACCELERATOR'] = 'cuda'
 

assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test_ipex.py

@@ -5,14 +5,14 @@
 import os
 import time
 from pathlib import Path
-from azure.ai.ml import command, Output, MLClient, PyTorchDistribution
+from azure.ai.ml import command, Output, MLClient
 from azure.ai.ml.entities import Environment, BuildContext, JobResourceConfiguration
 from azure.identity import AzureCliCredential
 import subprocess
 
 BUILD_CONTEXT = Path("../context")
 JOB_SOURCE_CODE = "../../acpt-tests/src"
-TIMEOUT_MINUTES = os.environ.get("timeout_minutes", 60)
+TIMEOUT_MINUTES = os.environ.get("timeout_minutes", 120)
 STD_LOG = Path("artifacts/user_logs/std_log.txt")
 
 
@@ -84,7 +84,4 @@ def test_pytorch_2_0():
  else:
  ml_client.jobs.stream(returned_job.name)
 
- assert current_status == "Completed"
-
-if __name__ == "__main__":
- test_pytorch_2_0()
+ assert current_status == "Completed"

assets/training/general/environments/tensorflow-2.12-cuda11/tests/tensorflow2_12_sample_test_cpu_bert.py

@@ -76,7 +76,4 @@ def test_tensorflow_2_12():
  else:
  ml_client.jobs.stream(returned_job.name)
 
- assert status == JobStatus.COMPLETED
-
-if __name__ == "__main__":
- test_tensorflow_2_12()
+ assert status == JobStatus.COMPLETED