From 6db74afd3ad25d52113df1e2d66f632202a7bf15 Mon Sep 17 00:00:00 2001 From: Ramin Azarmehr Date: Tue, 14 Feb 2023 17:02:14 -0500 Subject: [PATCH] Remove torch._six from test_mps --- test/test_mps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_mps.py b/test/test_mps.py index 2085d0cebe721..b7907e7ed1990 100644 --- a/test/test_mps.py +++ b/test/test_mps.py @@ -19,7 +19,7 @@ import yaml import platform from collections import defaultdict -from torch._six import inf +from torch import inf from torch.nn import Parameter from torch.testing._internal import opinfo from torch.testing._internal.common_utils import \