Preserve the order of candidates for pg.typing.Union.

PiperOrigin-RevId: 620082515
google · Mar 28, 2024 · 1fbcb8a · 1fbcb8a
1 parent e15b9a2
commit 1fbcb8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyglove/core/typing/annotation_conversion.py b/pyglove/core/typing/annotation_conversion.py
@@ -171,7 +171,7 @@ def _value_spec_from_type_annotation(
     if len(args) == 1:
       spec = _value_spec_from_annotation(args[0], True)
     else:
-      spec = vs.Union([_value_spec_from_annotation(x, True) for x in set(args)])
+      spec = vs.Union([_value_spec_from_annotation(x, True) for x in args])
     if optional:
       spec = spec.noneable()
     return spec