Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix deprecated call (#6005)

This commit is contained in:
Jirka Borovec 2021-02-16 12:11:13 +01:00 committed by GitHub
parent 6dba26666a
commit e6a2ccc850
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utilities/test_xla_device_utils.py
View File

@ -16,7 +16,7 @@ from unittest.mock import patch
import pytest
import pytorch_lightning.utilities.xla_device_utils as xla_utils
import pytorch_lightning.utilities.xla_device as xla_utils
from pytorch_lightning.utilities import _TPU_AVAILABLE, _XLA_AVAILABLE
from tests.helpers.utils import pl_multi_process_test
@ -34,7 +34,7 @@ def test_tpu_device_presence():
assert xla_utils.XLADeviceUtils.tpu_device_exists() is True
@patch('pytorch_lightning.utilities.xla_device_utils.TPU_CHECK_TIMEOUT', 10)
@patch('pytorch_lightning.utilities.xla_device.TPU_CHECK_TIMEOUT', 10)
def test_result_returns_within_timeout_seconds():
"""Check that pl_multi_process returns within 10 seconds"""
start = time.time()