remove dependency for dtype getters (#4291)

84888b93 · Philip Meier · GitHub · 759c5b6d · 84888b93
Unverified Commit 84888b93 authored Aug 19, 2021 by Philip Meier Committed by GitHub Aug 19, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/common_utils.py test/common_utils.py +2 -2

No files found.
--- a/test/common_utils.py
+++ b/test/common_utils.py
@@ -102,11 +102,11 @@ def cycle_over(objs):


 def int_dtypes():
-    return torch.testing.integral_types()
+    return (torch.uint8, torch.int8, torch.int16, torch.int32, torch.int64)


 def float_dtypes():
-    return torch.testing.floating_types()
+    return (torch.float32, torch.float64)


 @contextlib.contextmanager