Exclude union from entrypoints

src/flyte/__init__.py

@@ -5,6 +5,7 @@
 from __future__ import annotations
 
 import sys
+from importlib import metadata
 
 from ._build import build
 from ._cache import Cache, CachePolicy, CacheRequest
@@ -28,6 +29,7 @@
 from ._version import __version__
 
 sys.excepthook = custom_excepthook
+_original_entry_points = metadata.entry_points
 
 
 def _silence_grpc_warnings():
@@ -48,6 +50,21 @@ def _silence_grpc_warnings():
         os.environ["ABSL_LOG"] = "0"
 
 
+def _filtered_entry_points(*args, **kwargs):
+    """Wrap importlib.metadata.entry_points to exclude a specific distribution."""
+    eps = _original_entry_points(*args, **kwargs)
+    excluded_distribution = ["union", "unionai"]
+
+    if isinstance(eps, dict):  # Python 3.10/3.11
+        return {
+            group: [ep for ep in group_eps if ep.name not in excluded_distribution]
+            for group, group_eps in eps.items()
+        }
+    # Python 3.12/3.13
+    return metadata.EntryPoints(ep for ep in eps if ep.dist is None or ep.dist.name not in excluded_distribution)
+
+
+metadata.entry_points = _filtered_entry_points
 _silence_grpc_warnings()