Address review: broaden exception catch to preserve lazy-failure contract

Author: vatsrahul1001

task-sdk/src/airflow/sdk/io/path.py

@@ -125,8 +125,8 @@ def __init__(
         if self._conn_id is not None:
             try:
                 self.__wrapped__._fs_cached = attach(self.protocol or "file", self._conn_id).fs
-            except (ImportError, ModuleNotFoundError):
-                pass  # Provider package not installed; will fail naturally when the path is used
+            except Exception:
+                pass  # Provider not available or scheme unregistered; will fail naturally when the path is used
 
     @classmethod_or_method  # type: ignore[arg-type]
     def _from_upath(cls_or_self, upath, /):
@@ -145,8 +145,8 @@ def _from_upath(cls_or_self, upath, /):
         if obj._conn_id is not None and not hasattr(upath, "_fs_cached"):
             try:
                 obj.__wrapped__._fs_cached = attach(upath.protocol or "file", obj._conn_id).fs
-            except (ImportError, ModuleNotFoundError):
-                pass  # Provider package not installed; will fail naturally when the path is used
+            except Exception:
+                pass  # Provider not available or scheme unregistered; will fail naturally when the path is used
         return obj
 
     @property

task-sdk/tests/task_sdk/io/test_path.py

@@ -303,7 +303,9 @@ def test_rmdir_uses_authenticated_fs(self, fake_fs_with_conn):
         """rmdir() must use self.fs (Airflow-attached) not __wrapped__.fs (unauthenticated)."""
         fake_fs_with_conn.mkdir("bucket/empty_dir")
         p = ObjectStoragePath("ffs2://my_conn@bucket/empty_dir", conn_id="my_conn")
-        p.rmdir(recursive=False)
+        # upath's rmdir(recursive=False) calls next(self.iterdir()) without a default,
+        # which raises StopIteration on empty dirs — a upath bug. Use the default (recursive=True).
+        p.rmdir()
         assert not fake_fs_with_conn.exists("bucket/empty_dir")
 
     def test_conn_id_in_uri_works_for_exists(self, fake_fs_with_conn):