flake8

Lightning-AI · Feb 1, 2021 · 6ef068a · 6ef068a
1 parent 4fda04a
commit 6ef068a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/test_lightning_module.py b/tests/core/test_lightning_module.py
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from unittest.mock import patch, Mock, PropertyMock
+from unittest.mock import patch, Mock
 
 import pytest
 from torch.optim import Adam, SGD