fix wait_for_deployment to also check for apps/v1 deployments (kubefl…

…ow#477)

Signed-off-by: Yannis Zarkadas <yanniszark@arrikto.com>

py/kubeflow/testing/util.py

@@ -410,9 +410,13 @@ def wait_for_deployment(api_client,
     minutes=timeout_minutes)
 
   ext_client = k8s_client.ExtensionsV1beta1Api(api_client)
+  ext_client_apps = k8s_client.AppsV1beta1Api(api_client)
 
   while datetime.datetime.now() < end_time:
-    deploy = ext_client.read_namespaced_deployment(name, namespace)
+    try:
+      deploy = ext_client.read_namespaced_deployment(name, namespace)
+    except rest.ApiException:
+      deploy = ext_client_apps.read_namespaced_deployment(name, namespace)
     # ready_replicas could be None
     if (deploy.status.ready_replicas and
         deploy.status.ready_replicas >= replicas):