[refactor] Modified collections to collections.abc

facebookresearch · May 13, 2022 · 0bcf4da · 0bcf4da
1 parent b33e705
commit 0bcf4da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/fairscale/optim/grad_scaler.py b/fairscale/optim/grad_scaler.py
@@ -3,7 +3,9 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
-from collections import abc, defaultdict
+import collections.abc as abc
+from collections import defaultdict
+
 from enum import Enum
 import logging
 from typing import Any, Dict, List, Optional, Union

diff --git a/fairscale/utils/params.py b/fairscale/utils/params.py
@@ -3,7 +3,7 @@
 # This source code is licensed under the BSD license found in the
 # LICENSE file in the root directory of this source tree.
 
-from collections import abc
+import collections.abc as abc
 from math import inf
 from typing import Any, Callable, Dict, List, Optional