Fix import

Author: SeanNaren

tests/deprecated_api/test_remove_1-5.py

@@ -15,6 +15,7 @@
 import operator
 import os
 from typing import Any, Dict
+from unittest import mock
 
 import pytest
 import torch