darabos commited on
Commit
20375b5
·
1 Parent(s): e44317d

Make all NetworkX ops async for progress indication.

Browse files
lynxkite-graph-analytics/src/lynxkite_graph_analytics/networkx_ops.py CHANGED
@@ -152,11 +152,11 @@ def types_from_doc(doc: str) -> dict[str, type]:
152
 
153
  def wrapped(name: str, func):
154
  @functools.wraps(func)
155
- def wrapper(*args, **kwargs):
156
  for k, v in kwargs.items():
157
  if v == "None":
158
  kwargs[k] = None
159
- res = func(*args, **kwargs)
160
  # Figure out what the returned value is.
161
  if isinstance(res, nx.Graph):
162
  return res
 
152
 
153
  def wrapped(name: str, func):
154
  @functools.wraps(func)
155
+ async def wrapper(*args, **kwargs):
156
  for k, v in kwargs.items():
157
  if v == "None":
158
  kwargs[k] = None
159
+ res = await ops.slow(func)(*args, **kwargs)
160
  # Figure out what the returned value is.
161
  if isinstance(res, nx.Graph):
162
  return res