提交 210984dd authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Replace defaultdict(int) with Counter in pkl_utils

上级 3b6b0649
...@@ -10,7 +10,7 @@ import pickle ...@@ -10,7 +10,7 @@ import pickle
import sys import sys
import tempfile import tempfile
import zipfile import zipfile
from collections import defaultdict from collections import Counter
from contextlib import closing from contextlib import closing
from io import BytesIO from io import BytesIO
from pickle import HIGHEST_PROTOCOL from pickle import HIGHEST_PROTOCOL
...@@ -144,7 +144,7 @@ class PersistentSharedVariableID(PersistentNdarrayID): ...@@ -144,7 +144,7 @@ class PersistentSharedVariableID(PersistentNdarrayID):
def __init__(self, zip_file, allow_unnamed=True, allow_duplicates=True): def __init__(self, zip_file, allow_unnamed=True, allow_duplicates=True):
super().__init__(zip_file) super().__init__(zip_file)
self.name_counter = defaultdict(int) self.name_counter = Counter()
self.ndarray_names = {} self.ndarray_names = {}
self.allow_unnamed = allow_unnamed self.allow_unnamed = allow_unnamed
self.allow_duplicates = allow_duplicates self.allow_duplicates = allow_duplicates
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论