Skip to content

Fix graph (pyg) file output name

Jared Dynes Burleson requested to merge burleson into dev

So I have noticed that graphs (.pyg) files when output from the first stage of the code (graph construction via mm or metric learning) following the form:

event000000XXX.pyg

However, when a second stage is run (for example the GNN training or the Filter training), the name of the file dumps out as:

event['\''000000XXX'\''].pyg

And if you keep running more stages on this, it keeps adding more of these single quotes and brackets, which are rather messy and cause issues when trying to read the file into python.

I have fixed it in the Edge_Classifier/Connected_Components/Walkthrough

Merge request reports