diff --git a/test/e2e/preset_test.go b/test/e2e/preset_test.go index ff424cb59..f4df8de22 100644 --- a/test/e2e/preset_test.go +++ b/test/e2e/preset_test.go @@ -581,171 +581,171 @@ var _ = Describe("Workspace Preset", func() { } }) - It("should create a mistral workspace with preset public mode successfully", func() { - numOfNode := 1 - workspaceObj := createMistralWorkspaceWithPresetPublicMode(numOfNode) + // It("should create a mistral workspace with preset public mode successfully", func() { + // numOfNode := 1 + // workspaceObj := createMistralWorkspaceWithPresetPublicMode(numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a Phi-2 workspace with preset public mode successfully", func() { - numOfNode := 1 - workspaceObj := createPhi2WorkspaceWithPresetPublicMode(numOfNode) + // It("should create a Phi-2 workspace with preset public mode successfully", func() { + // numOfNode := 1 + // workspaceObj := createPhi2WorkspaceWithPresetPublicMode(numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a falcon workspace with preset public mode successfully", func() { - numOfNode := 1 - workspaceObj := createFalconWorkspaceWithPresetPublicMode(numOfNode) + // It("should create a falcon workspace with preset public mode successfully", func() { + // numOfNode := 1 + // workspaceObj := createFalconWorkspaceWithPresetPublicMode(numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a llama 7b workspace with preset private mode successfully", func() { - numOfNode := 1 - modelVersion, ok := modelInfo[PresetLlama2AChat] - if !ok { - Fail(fmt.Sprintf("Model version for %s not found", PresetLlama2AChat)) - } - workspaceObj := createLlama7BWorkspaceWithPresetPrivateMode(aiModelsRegistry, aiModelsRegistrySecret, modelVersion, numOfNode) + // It("should create a llama 7b workspace with preset private mode successfully", func() { + // numOfNode := 1 + // modelVersion, ok := modelInfo[PresetLlama2AChat] + // if !ok { + // Fail(fmt.Sprintf("Model version for %s not found", PresetLlama2AChat)) + // } + // workspaceObj := createLlama7BWorkspaceWithPresetPrivateMode(aiModelsRegistry, aiModelsRegistrySecret, modelVersion, numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a llama 13b workspace with preset private mode successfully", func() { - if !runLlama13B { - Skip("Skipping llama 13b workspace test") - } - numOfNode := 2 - modelVersion, ok := modelInfo[PresetLlama2BChat] - if !ok { - Fail(fmt.Sprintf("Model version for %s not found", PresetLlama2AChat)) - } - workspaceObj := createLlama13BWorkspaceWithPresetPrivateMode(aiModelsRegistry, aiModelsRegistrySecret, modelVersion, numOfNode) + // It("should create a llama 13b workspace with preset private mode successfully", func() { + // if !runLlama13B { + // Skip("Skipping llama 13b workspace test") + // } + // numOfNode := 2 + // modelVersion, ok := modelInfo[PresetLlama2BChat] + // if !ok { + // Fail(fmt.Sprintf("Model version for %s not found", PresetLlama2AChat)) + // } + // workspaceObj := createLlama13BWorkspaceWithPresetPrivateMode(aiModelsRegistry, aiModelsRegistrySecret, modelVersion, numOfNode) - defer cleanupResources(workspaceObj) + // defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // time.Sleep(30 * time.Second) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), true) + // validateInferenceResource(workspaceObj, int32(numOfNode), true) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a custom template workspace successfully", func() { - numOfNode := 1 - imageName := "nginx:latest" - workspaceObj := createCustomWorkspaceWithPresetCustomMode(imageName, numOfNode) + // It("should create a custom template workspace successfully", func() { + // numOfNode := 1 + // imageName := "nginx:latest" + // workspaceObj := createCustomWorkspaceWithPresetCustomMode(imageName, numOfNode) - defer cleanupResources(workspaceObj) + // defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // time.Sleep(30 * time.Second) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a Phi-3-mini-128k-instruct workspace with preset public mode successfully", func() { - numOfNode := 1 - workspaceObj := createPhi3WorkspaceWithPresetPublicMode(numOfNode) + // It("should create a Phi-3-mini-128k-instruct workspace with preset public mode successfully", func() { + // numOfNode := 1 + // workspaceObj := createPhi3WorkspaceWithPresetPublicMode(numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateAssociatedService(workspaceObj) + // validateAssociatedService(workspaceObj) - validateInferenceResource(workspaceObj, int32(numOfNode), false) + // validateInferenceResource(workspaceObj, int32(numOfNode), false) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) - It("should create a workspace for tuning successfully", func() { - numOfNode := 1 - err := copySecretToNamespace(aiModelsRegistrySecret, namespaceName) - if err != nil { - log.Fatalf("Error copying secret: %v", err) - } - configMap := createCustomTuningConfigMapForE2E() - workspaceObj, jobName := createPhi3TuningWorkspaceWithPresetPublicMode(configMap.Name, numOfNode) + // It("should create a workspace for tuning successfully", func() { + // numOfNode := 1 + // err := copySecretToNamespace(aiModelsRegistrySecret, namespaceName) + // if err != nil { + // log.Fatalf("Error copying secret: %v", err) + // } + // configMap := createCustomTuningConfigMapForE2E() + // workspaceObj, jobName := createPhi3TuningWorkspaceWithPresetPublicMode(configMap.Name, numOfNode) - defer cleanupResources(workspaceObj) - time.Sleep(30 * time.Second) + // defer cleanupResources(workspaceObj) + // time.Sleep(30 * time.Second) - validateMachineCreation(workspaceObj, numOfNode) - validateResourceStatus(workspaceObj) + // validateMachineCreation(workspaceObj, numOfNode) + // validateResourceStatus(workspaceObj) - time.Sleep(30 * time.Second) + // time.Sleep(30 * time.Second) - validateTuningResource(workspaceObj) + // validateTuningResource(workspaceObj) - validateACRTuningResultsUploaded(workspaceObj, jobName) + // validateACRTuningResultsUploaded(workspaceObj, jobName) - validateWorkspaceReadiness(workspaceObj) - }) + // validateWorkspaceReadiness(workspaceObj) + // }) })