teaching/muMLE
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Updated types

Browse source
This commit is contained in:
Andrei Bondarenko 2021-07-14 08:41:08 +02:00
parent 24119609b3
commit 93b602a0eb
3 changed files with 18 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

16
state/base.py
View file

@ -3,13 +3,13 @@ from typing import Any, List, Tuple, Optional, Union
import uuid
primitive_types = (int, float, str, bool)
INTEGER = {"type": "Integer"}
FLOAT = {"type": "Float"}
STRING = {"type": "String"}
BOOLEAN = {"type": "Boolean"}
TYPE = {"type": "Type"}
NODE = {"type": "Node"}
EDGE = {"type": "Edge"}
INTEGER = ("Integer",)
FLOAT = ("Float",)
STRING = ("String",)
BOOLEAN = ("Boolean",)
TYPE = ("Type",)
NODE = ("Node",)
EDGE = ("Edge",)
type_values = (INTEGER, FLOAT, STRING, BOOLEAN, TYPE, NODE, EDGE)
@ -44,7 +44,7 @@ class State(ABC):
Returns:
True if value type is supported, False otherwise.
"""
if isinstance(value, dict) and value in type_values:
if isinstance(value, tuple) and value in type_values:
return True
if not isinstance(value, primitive_types):
return False