Skip to content

Commit

Permalink
retrigger issue
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaansehgal99 committed Jul 11, 2024
1 parent eb0b4f5 commit 264b9e1
Showing 1 changed file with 112 additions and 112 deletions.
224 changes: 112 additions & 112 deletions test/e2e/preset_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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)
// })

})

0 comments on commit 264b9e1

Please sign in to comment.