3.9 fix

Author: CalebCourier

tests/integration_tests/test_formatters.py

@@ -1,6 +1,6 @@
 import importlib
 import pytest
-from typing import List, Union
+from typing import List
 
 from pydantic import BaseModel
 
@@ -78,5 +78,7 @@ class Tricky(BaseModel):
     assert isinstance(out, dict)
     assert "foo" in out
     assert isinstance(out["foo"], dict)
-    assert isinstance(out["foo"]["whole"], Union[int, float])
+    assert isinstance(out["foo"]["whole"], int) or isinstance(
+        out["foo"]["whole"], float
+    )
     assert isinstance(out["foo"]["frac"], float)