From 7b9c851ab7030cbe03ae667e684c5af57623d2d8 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Thu, 19 Aug 2021 10:08:48 +0200 Subject: [PATCH] remove dependency for dtype getters --- test/common_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/common_utils.py b/test/common_utils.py index 79fce27110d..1a2e3042365 100644 --- 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