Additional review comments - moved result to score base class

Signed-off-by: Maya Anderson <mayaa@il.ibm.com>
This commit is contained in:
Maya Anderson 2023-03-13 00:30:18 +02:00
parent 4988fea08c
commit 8a4df5a4a2
4 changed files with 11 additions and 15 deletions

View file

@ -1,23 +1,20 @@
from dataclasses import dataclass, field
from typing import Optional
import numpy as np
DEFAULT_DATASET_NAME = "dataset"
@dataclass
class DatasetAttackScore:
dataset_name: str
@dataclass
class DatasetAttackResult:
pass
@dataclass(repr=False)
class DatasetAttackScoreWithResult(DatasetAttackScore):
result: DatasetAttackResult = field(repr=False)
@dataclass
class DatasetAttackScore:
dataset_name: str
result: Optional[DatasetAttackResult] = None
@dataclass