Skip to content

Commit 556e13e

Browse files
committed
Changes from Shankha Nag
1 parent 984be0f commit 556e13e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyironflow/wf_extensions.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from pyiron_workflow.type_hinting import type_hint_to_tuple, valid_value
2-
from pyiron_workflow.channels import NotData
2+
from pyiron_workflow.api import NOT_DATA
33
from pyiron_workflow.node import Node
44
from pyironflow.themes import get_color
55
import importlib
@@ -54,7 +54,7 @@ def dict_to_node(dict_node: dict, live_children: dict = None, reload=False) -> N
5454
target_values = data['target_values']
5555
target_labels = data['target_labels']
5656
for k, v in zip(target_labels, target_values):
57-
if v not in ('NonPrimitive', 'NotData', ''):
57+
if v not in ('NonPrimitive', 'NOT_DATA.__class__', ''):
5858
type_hint = node.inputs[k].type_hint
5959
# JS gui can return input values like 2.0 as int, breaking type hints
6060
# so check here if the type hint is a float, but convert only if losslessly possible
@@ -80,8 +80,8 @@ def get_node_values(channel_dict):
8080
values = []
8181
for k, v in channel_dict.items():
8282
value = v.value
83-
if isinstance(value, NotData):
84-
value = 'NotData'
83+
if isinstance(value, NOT_DATA.__class__):
84+
value = 'NOT_DATA.__class__'
8585
elif not is_primitive(value):
8686
value = 'NonPrimitive'
8787

0 commit comments

Comments
 (0)