Skip to content

Commit 31cf068

Browse files
committed
Fixes
1 parent 40a8ebf commit 31cf068

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
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: 1 addition & 1 deletion
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

0 commit comments

Comments
 (0)