From 5f08593c1a2b65f88b434f849d4c923150b7fa66 Mon Sep 17 00:00:00 2001
From: brentyi <yibrenth@gmail.com>
Date: Sun, 24 Nov 2024 02:32:11 -0800
Subject: [PATCH] UnionType import fix

---
 src/viser/infra/_typescript_interface_gen.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/viser/infra/_typescript_interface_gen.py b/src/viser/infra/_typescript_interface_gen.py
index 2c939ba8a..2bba44ca5 100644
--- a/src/viser/infra/_typescript_interface_gen.py
+++ b/src/viser/infra/_typescript_interface_gen.py
@@ -1,6 +1,6 @@
 import dataclasses
+import types
 from collections import defaultdict
-from types import UnionType
 from typing import Any, Type, Union, cast
 
 import numpy as np
@@ -49,6 +49,7 @@ def _get_ts_type(typ: Type[Any]) -> str:
         origin_typ = args[0]
 
     # Automatic Python => TypeScript conversion.
+    UnionType = getattr(types, "UnionType", Union)
     if origin_typ is tuple:
         args = get_args(typ)
         if len(args) == 2 and args[1] == ...: