提交 c789bf31 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!97 Append parameter verification for 'attack_config'

Merge pull request !97 from liuluobin/master
...@@ -124,7 +124,14 @@ class MembershipInference: ...@@ -124,7 +124,14 @@ class MembershipInference:
raise TypeError("Type of parameter 'test_train' must be Dataset, " raise TypeError("Type of parameter 'test_train' must be Dataset, "
"but got {}".format(type(dataset_train))) "but got {}".format(type(dataset_train)))
if not isinstance(attack_config, list):
raise TypeError("Type of parameter 'attack_config' must be list, "
"but got {}.".format(type(attack_config)))
for config in attack_config: for config in attack_config:
if not isinstance(config, dict):
raise TypeError("Type of each config in 'attack_config' must be dict, "
"but got {}.".format(type(config)))
if {"params", "method"} != set(config.keys()): if {"params", "method"} != set(config.keys()):
raise KeyError("Each config in attack_config must have keys 'method' and 'params', " raise KeyError("Each config in attack_config must have keys 'method' and 'params', "
"but your key value is {}.".format(set(config.keys()))) "but your key value is {}.".format(set(config.keys())))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册