Skip to content

Commit 8ce7055

Browse files
authored
[Fix] Fix mmengine dump of tversky ut in dev-1.x (#2001)
1 parent ee7f340 commit 8ce7055

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_models/test_losses/test_tversky_loss.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,12 @@ def test_tversky_lose():
3636
import os
3737
import tempfile
3838

39-
import mmcv
39+
import mmengine
4040
import numpy as np
4141
tmp_file = tempfile.NamedTemporaryFile()
4242

43-
mmcv.dump([1.0, 2.0, 3.0], f'{tmp_file.name}.pkl', 'pkl') # from pkl file
43+
mmengine.dump([1.0, 2.0, 3.0], f'{tmp_file.name}.pkl',
44+
'pkl') # from pkl file
4445
loss_cfg = dict(
4546
type='TverskyLoss',
4647
class_weight=f'{tmp_file.name}.pkl',

0 commit comments

Comments
 (0)