summaryrefslogtreecommitdiff
path: root/comments.py
diff options
context:
space:
mode:
Diffstat (limited to 'comments.py')
-rw-r--r--comments.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/comments.py b/comments.py
index afbcbf0..98c80c7 100644
--- a/comments.py
+++ b/comments.py
@@ -11,11 +11,15 @@ a separate base
import sys
import copy
-from collections import MutableSet, Sized, Set
from ruamel.yaml.compat import ordereddict, PY2, string_types
from ruamel.yaml.scalarstring import ScalarString
+if PY2:
+ from collections import MutableSet, Sized, Set
+else:
+ from collections.abc import MutableSet, Sized, Set
+
if False: # MYPY
from typing import Any, Dict, Optional, List, Union # NOQA