diff --git a/ax/models/torch/tests/test_model.py b/ax/models/torch/tests/test_model.py
index 1cb2ec91381..55b50d385c0 100644
--- a/ax/models/torch/tests/test_model.py
+++ b/ax/models/torch/tests/test_model.py
@@ -5,7 +5,6 @@
 # LICENSE file in the root directory of this source tree.
 
 import dataclasses
-import warnings
 from contextlib import ExitStack
 from copy import deepcopy
 from typing import Dict, OrderedDict, Type