Spaces:
Running
Running
File size: 1,322 Bytes
0c44583 d994c06 0c44583 d994c06 0c44583 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""Automatically wraps all NetworkX functions as LynxKite operations."""
from . import ops
import functools
import inspect
import networkx as nx
def wrapped(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
for k, v in kwargs.items():
if v == 'None':
kwargs[k] = None
res = func(*args, **kwargs)
if isinstance(res, nx.Graph):
return res
# Otherwise it's a node attribute.
graph = args[0].copy()
nx.set_node_attributes(graph, 'attr', name)
return graph
return wrapper
for (name, func) in nx.__dict__.items():
if hasattr(func, 'graphs'):
sig = inspect.signature(func)
inputs = {k: nx.Graph for k in func.graphs}
params = {
name:
str(param.default)
if type(param.default) in [str, int, float]
else None
for name, param in sig.parameters.items()
if name not in ['G', 'backend', 'backend_kwargs']}
for k, v in params.items():
if sig.parameters[k].annotation is inspect._empty and v is None:
# No annotation, no default — we must guess the type.
if len(k) == 1:
params[k] = 1
name = "NX › " + name.replace('_', ' ').title()
op = ops.Op(wrapped(func), name, params=params, inputs=inputs, outputs={'output': 'yes'}, type='basic')
ops.ALL_OPS[name] = op
|