Convert VariableFeatures into dataclass

This commit is contained in:
2021-04-04 22:56:26 -05:00
parent 59f4f75a53
commit d79eec5da6
6 changed files with 97 additions and 87 deletions

View File

@@ -87,14 +87,12 @@ InstanceFeatures = TypedDict(
total=False,
)
VariableFeatures = TypedDict(
"VariableFeatures",
{
"Category": Optional[Hashable],
"User features": Optional[List[float]],
},
total=False,
)
@dataclass
class VariableFeatures:
category: Optional[Hashable] = None
user_features: Optional[List[float]] = None
ConstraintFeatures = TypedDict(
"ConstraintFeatures",