darabos commited on
Commit
f1ce4c4
·
1 Parent(s): 1df031f

Fix some more NetworkX operation names.

Browse files
lynxkite-graph-analytics/src/lynxkite_graph_analytics/networkx_ops.py CHANGED
@@ -218,29 +218,66 @@ def _get_params(func) -> dict | None:
218
 
219
 
220
  _REPLACEMENTS = [
221
- ("Barabasi Albert", "Barabasi–Albert"),
222
- ("Bellman Ford", "Bellman–Ford"),
223
- ("Bethe Hessian", "Bethe–Hessian"),
224
- ("Bfs", "BFS"),
225
- ("Dag ", "DAG "),
226
- ("Dfs", "DFS"),
227
- ("Dorogovtsev Goltsev Mendes", "DorogovtsevGoltsev–Mendes"),
228
- ("Erdos Renyi", "ErdosRenyi"),
229
- ("Floyd Warshall", "FloydWarshall"),
230
- ("Gnc", "G(n,c)"),
231
- ("Gnm", "G(n,m)"),
232
- ("Gnp", "G(n,p)"),
233
- ("Gnr", "G(n,r)"),
234
- ("Havel Hakimi", "Havel–Hakimi"),
235
- ("Hkn", "H(k,n)"),
236
- ("Hnm", "H(n,m)"),
237
- ("Kl ", "KL "),
238
- ("Moebius Kantor", "Moebius–Kantor"),
239
- ("Pagerank", "PageRank"),
240
- ("Scale Free", "Scale-Free"),
241
- ("Vf2Pp", "VF2++"),
242
- ("Watts Strogatz", "Watts–Strogatz"),
243
- ("Weisfeiler Lehman", "Weisfeiler–Lehman"),
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
244
  ]
245
  _CATEGORY_REPLACEMENTS = [
246
  ("Networkx", "NetworkX"),
@@ -289,9 +326,11 @@ def register_networkx(env: str):
289
  except UnsupportedParameterType:
290
  continue
291
  inputs = [ops.Input(name=k, type=nx.Graph) for k in func.graphs]
292
- nicename = name.replace("_", " ").title()
293
  for a, b in _REPLACEMENTS:
294
  nicename = nicename.replace(a, b)
 
 
295
  op = ops.Op(
296
  func=wrapped(name, func),
297
  name=nicename,
 
218
 
219
 
220
  _REPLACEMENTS = [
221
+ (" at free", " AT-free"),
222
+ (" dag", " DAG"),
223
+ (" k out ", " k-out "),
224
+ (" rary", " r-ary"),
225
+ ("2d ", "2D "),
226
+ ("3d ", "3D "),
227
+ ("adamic adar", "AdamicAdar"),
228
+ ("barabasi albert", "BarabasiAlbert"),
229
+ ("bellman ford", "BellmanFord"),
230
+ ("bethe hessian", "Bethe–Hessian"),
231
+ ("bfs", "BFS"),
232
+ ("d separator", "d-separator"),
233
+ ("dag ", "DAG "),
234
+ ("dfs", "DFS"),
235
+ ("dijkstra", "Dijkstra"),
236
+ ("dorogovtsev goltsev mendes", "Dorogovtsev–Goltsev–Mendes"),
237
+ ("erdos renyi", "Erdos–Renyi"),
238
+ ("euler", "Euler"),
239
+ ("floyd warshall", "Floyd–Warshall"),
240
+ ("forceatlas2", "ForceAtlas2"),
241
+ ("gexf ", "GEXF "),
242
+ ("gml", "GML"),
243
+ ("gnc", "G(n,c)"),
244
+ ("gnm", "G(n,m)"),
245
+ ("gnp", "G(n,p)"),
246
+ ("gnr", "G(n,r)"),
247
+ ("graphml", "GraphML"),
248
+ ("harary", "Harary"),
249
+ ("havel hakimi", "Havel–Hakimi"),
250
+ ("hkn", "H(k,n)"),
251
+ ("hnm", "H(n,m)"),
252
+ ("internet", "Internet"),
253
+ ("k core", "k-core"),
254
+ ("k corona", "k-corona"),
255
+ ("k crust", "k-crust"),
256
+ ("k shell", "k-shell"),
257
+ ("k truss", "k-truss"),
258
+ ("kl ", "KL "),
259
+ ("laplacian", "Laplacian"),
260
+ ("lfr ", "LFR "),
261
+ ("margulis gabber galil", "Margulis–Gabber–Galil"),
262
+ ("moebius kantor", "Moebius–Kantor"),
263
+ ("newman watts strogatz", "Newman–Watts–Strogatz"),
264
+ ("numpy", "NumPy"),
265
+ ("pagerank", "PageRank"),
266
+ ("pajek", "Pajek"),
267
+ ("pandas", "Pandas"),
268
+ ("parse leda", "Parse LEDA"),
269
+ ("powerlaw", "power-law"),
270
+ ("prufer", "Prüfer"),
271
+ ("radzik", "Radzik"),
272
+ ("s metric", "s-metric"),
273
+ ("scale free", "Scale-free"),
274
+ ("scipy", "SciPy"),
275
+ ("small world", "small-world"),
276
+ ("soundarajan hopcroft", "Soundarajan–Hopcroft"),
277
+ ("southern women", "Southern women"),
278
+ ("vf2pp", "VF2++"),
279
+ ("watts strogatz", "Watts–Strogatz"),
280
+ ("weisfeiler lehman", "Weisfeiler–Lehman"),
281
  ]
282
  _CATEGORY_REPLACEMENTS = [
283
  ("Networkx", "NetworkX"),
 
326
  except UnsupportedParameterType:
327
  continue
328
  inputs = [ops.Input(name=k, type=nx.Graph) for k in func.graphs]
329
+ nicename = name.replace("_", " ")
330
  for a, b in _REPLACEMENTS:
331
  nicename = nicename.replace(a, b)
332
+ if nicename[1] != "-":
333
+ nicename = nicename[0].upper() + nicename[1:]
334
  op = ops.Op(
335
  func=wrapped(name, func),
336
  name=nicename,