build: fix e2e notebook cluster check

microsoft · Jul 30, 2019 · ce1e66d · ce1e66d
1 parent 19aeb80
commit ce1e66d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/it/scala/com/microsoft/ml/nbtest/NotebookTests.scala b/src/it/scala/com/microsoft/ml/nbtest/NotebookTests.scala
@@ -24,11 +24,14 @@ class NotebookTests extends TestBase {
 
   test("Databricks Notebooks") {
     tryWithRetries(Array.fill(500)(10000)) {() =>
+      println("Checking Cluster Availibility...")
       assert(listActiveJobs(clusterId).isEmpty,
         "Cluster already has running jobs cannot change libraries safely")
+      assert(listInstalledLibraries(clusterId).isEmpty,
+        "Cluster already has libraries installed")
+      println("Cluster now Availible")
     }
     try {
-      assert(listInstalledLibraries(clusterId).isEmpty, "Cluster already has libraries installed")
       println("Installing libraries")
       installLibraries(clusterId)
       println(s"Creating folder $folder")