CI: fix examples - patch download MNIST (#6357)

* patch download * CI * isort * extra remove runif
Lightning-AI · Mar 9, 2021 · bccd0b8 · bccd0b8
1 parent 4abf247
commit bccd0b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pytorch_lightning/plugins/training_type/training_type_plugin.py b/pytorch_lightning/plugins/training_type/training_type_plugin.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from abc import ABC, abstractmethod
-from typing import Any, Callable, Iterable, Optional, TYPE_CHECKING, Union
+from typing import Any, Callable, Dict, Iterable, Optional, TYPE_CHECKING, Union
 
 import torch
 from torch.nn import Module

diff --git a/tests/plugins/test_sharded_plugin.py b/tests/plugins/test_sharded_plugin.py
@@ -31,10 +31,13 @@ def test_ddp_sharded_precision_16_clip_gradients(mock_oss_clip_grad_norm, clip_v
         mock_oss_clip_grad_norm.assert_not_called()
 
 
+<<<<<<< HEAD
 <<<<<<< HEAD
 =======
 @RunIf(fairscale=True)
 >>>>>>> d0e7ca36... [Fix] Call clip gradients if clip val greater than 0 (#6330)
+=======
+>>>>>>> 48482da3... CI: fix examples - patch download MNIST (#6357)
 @pytest.mark.parametrize(["accelerator"], [("ddp_sharded", ), ("ddp_sharded_spawn", )])
 @pytest.mark.skipif(not _FAIRSCALE_AVAILABLE, reason="Fairscale is not available")
 def test_sharded_ddp_choice(tmpdir, accelerator):