Spaces:
Running
Running
Better error messages for wrong types in graph analytics workspaces.
Browse files
lynxkite-graph-analytics/src/lynxkite_graph_analytics/core.py
CHANGED
@@ -224,12 +224,16 @@ async def _execute_node(
|
|
224 |
missing.append(p.name)
|
225 |
continue
|
226 |
x = input_map[p.name]
|
227 |
-
if p.type == nx.Graph
|
228 |
-
x
|
229 |
-
|
230 |
-
x
|
231 |
-
elif p.type == Bundle
|
232 |
-
x
|
|
|
|
|
|
|
|
|
233 |
inputs.append(x)
|
234 |
except Exception as e:
|
235 |
if not os.environ.get("LYNXKITE_SUPPRESS_OP_ERRORS"):
|
|
|
224 |
missing.append(p.name)
|
225 |
continue
|
226 |
x = input_map[p.name]
|
227 |
+
if p.type == nx.Graph:
|
228 |
+
if isinstance(x, Bundle):
|
229 |
+
x = x.to_nx()
|
230 |
+
assert isinstance(x, nx.Graph), "Input must be a graph."
|
231 |
+
elif p.type == Bundle:
|
232 |
+
if isinstance(x, nx.Graph):
|
233 |
+
x = Bundle.from_nx(x)
|
234 |
+
elif isinstance(x, pd.DataFrame):
|
235 |
+
x = Bundle.from_df(x)
|
236 |
+
assert isinstance(x, Bundle), "Input must be a graph or dataframe."
|
237 |
inputs.append(x)
|
238 |
except Exception as e:
|
239 |
if not os.environ.get("LYNXKITE_SUPPRESS_OP_ERRORS"):
|