Update test/contrib/hsgp/test_approximation.py

Co-authored-by: Juan Orduz <[email protected]>
pyro-ppl · Jun 25, 2024 · 6982850 · 6982850
1 parent 1d16f4e
commit 6982850
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/contrib/hsgp/test_approximation.py b/test/contrib/hsgp/test_approximation.py
@@ -144,7 +144,7 @@ def test_kernel_approx_squared_exponential(
     def _exact_rbf(length):
         return RBF(length)(x1, x2).squeeze(axis=-1)
 
-    if isinstance(length, Union[float, int]):
+    if isinstance(length,  int) | isinstance(length, float):
         exact = _exact_rbf(length)
     elif length.ndim == 1:
         exact = _exact_rbf(length)