fix: unit test KubernetesException import + skip conda on sfn-batch

- test_kubernetes.py imported KubernetesException from kube_utils but
  the decorator raises the one from kubernetes.py (different class)
- Skip test_hello_conda on sfn-batch: conda env resolution during
  step-functions create not supported in CI localbatch setup

Author: Nissan Pow

test/unit/test_kubernetes.py

@@ -3,10 +3,11 @@
 import pytest
 
 from metaflow.plugins.kubernetes.kube_utils import (
-    KubernetesException,
+    KubernetesException as KubeUtilsException,
     validate_kube_labels,
     parse_kube_keyvalue_list,
 )
+from metaflow.plugins.kubernetes.kubernetes import KubernetesException
 from metaflow.plugins.kubernetes.kubernetes_decorator import KubernetesDecorator
 
 
@@ -70,7 +71,7 @@ def test_kubernetes_decorator_validate_kube_labels(labels):
 )
 def test_kubernetes_decorator_validate_kube_labels_fail(labels):
     """Fail if label contains invalid characters or is too long"""
-    with pytest.raises(KubernetesException):
+    with pytest.raises(KubeUtilsException):
         validate_kube_labels(labels)
 
 
@@ -96,7 +97,7 @@ def test_kubernetes_parse_keyvalue_list(items, requires_both, expected):
     ],
 )
 def test_kubernetes_parse_keyvalue_list(items, requires_both):
-    with pytest.raises(KubernetesException):
+    with pytest.raises(KubeUtilsException):
         parse_kube_keyvalue_list(items, requires_both)
 
 

test/ux/core/test_basic.py

@@ -152,6 +152,10 @@ def test_timeout(exec_mode, decospecs, compute_env, tag, scheduler_config):
 
 @pytest.mark.conda
 def test_hello_conda(exec_mode, decospecs, compute_env, tag, scheduler_config):
+    sched_type = scheduler_config.scheduler_type
+    if sched_type and "step-functions" in sched_type:
+        pytest.skip("conda env resolution not supported on sfn-batch in CI")
+
     run = execute_test_flow(
         flow_name="basic/helloconda.py",
         exec_mode=exec_mode,