Skip to content

Commit bccd0b8

Browse files
Bordatchaton
authored andcommitted
CI: fix examples - patch download MNIST (#6357)
* patch download * CI * isort * extra remove runif
1 parent 4abf247 commit bccd0b8

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

pytorch_lightning/plugins/training_type/training_type_plugin.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
from abc import ABC, abstractmethod
15-
from typing import Any, Callable, Iterable, Optional, TYPE_CHECKING, Union
15+
from typing import Any, Callable, Dict, Iterable, Optional, TYPE_CHECKING, Union
1616

1717
import torch
1818
from torch.nn import Module

tests/plugins/test_sharded_plugin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,13 @@ def test_ddp_sharded_precision_16_clip_gradients(mock_oss_clip_grad_norm, clip_v
3131
mock_oss_clip_grad_norm.assert_not_called()
3232

3333

34+
<<<<<<< HEAD
3435
<<<<<<< HEAD
3536
=======
3637
@RunIf(fairscale=True)
3738
>>>>>>> d0e7ca36... [Fix] Call clip gradients if clip val greater than 0 (#6330)
39+
=======
40+
>>>>>>> 48482da3... CI: fix examples - patch download MNIST (#6357)
3841
@pytest.mark.parametrize(["accelerator"], [("ddp_sharded", ), ("ddp_sharded_spawn", )])
3942
@pytest.mark.skipif(not _FAIRSCALE_AVAILABLE, reason="Fairscale is not available")
4043
def test_sharded_ddp_choice(tmpdir, accelerator):

0 commit comments

Comments
 (0)