Skip to content

Commit 58046c8

Browse files
committed
Fixes
1 parent 5328277 commit 58046c8

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/profiler/test_profiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def test_simple_profiler_logs(tmpdir, caplog, simple_profiler):
147147
"""Ensure that the number of printed logs is correct."""
148148
model = BoringModel()
149149
trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=2, profiler=simple_profiler, logger=False)
150-
with caplog.at_level(logging.INFO, logger="pytorch_lightning.profiler.profilers"):
150+
with caplog.at_level(logging.INFO, logger="pytorch_lightning.profiler"):
151151
trainer.fit(model)
152152
trainer.test(model)
153153

tests/trainer/test_supporters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414
import os
15-
from collections import Sequence
15+
from typing import Sequence
1616
from unittest import mock
1717

1818
import pytest
@@ -349,7 +349,7 @@ def __init__(self, data_source, name) -> None:
349349

350350
with mock.patch.dict(os.environ, {"PL_FAULT_TOLERANT_TRAINING": str(int(use_fault_tolerant))}):
351351

352-
trainer = Trainer(replace_sampler_ddp=replace_sampler_ddp, accelerator="ddp", gpus=2)
352+
trainer = Trainer(replace_sampler_ddp=replace_sampler_ddp, strategy="ddp", gpus=2)
353353
dataloader = trainer.prepare_dataloader(dataloader, shuffle=True)
354354
_count = 0
355355
_has_fastforward_sampler = False

0 commit comments

Comments
 (0)