626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Make this code compatible with Python 3.10.

Browse source 
PiperOrigin-RevId: 473313795
This commit is contained in:
Yilei Yang 2022-09-09 12:19:14 -07:00 committed by A. Unique TensorFlower
parent 715fd1a670
commit ebae6c086e
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/dataset_slicing.py
View file

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
"""Specifying and creating AttackInputData slices.""" """Specifying and creating AttackInputData slices."""
import collections from collections import abc
import copy import copy
import logging import logging
from typing import List, Optional from typing import List, Optional
@ -143,7 +143,7 @@ def get_single_slice_specs(
result.append(SingleSliceSpec(SlicingFeature.CLASS, c)) result.append(SingleSliceSpec(SlicingFeature.CLASS, c))
elif isinstance(by_class, int): elif isinstance(by_class, int):
result.append(SingleSliceSpec(SlicingFeature.CLASS, by_class)) result.append(SingleSliceSpec(SlicingFeature.CLASS, by_class))
elif isinstance(by_class, collections.Iterable): elif isinstance(by_class, abc.Iterable):
for c in by_class: for c in by_class:
result.append(SingleSliceSpec(SlicingFeature.CLASS, c)) result.append(SingleSliceSpec(SlicingFeature.CLASS, c))