[BUG]: Missing map_node_id in GNN Fraud Detection training notebook

[efajardo-nv]

Version

23.11

Which installation method(s) does this occur on?

Docker

Describe the bug.

An ImportError occurs in the first cell of the GNN Fraud Detection training notebook because map_node_id does not exist in training.py.

Minimum reproducible example

1. cd to models/training-tuning-scripts/fraud-detection-models

2. Create or update conda environment with required dependencies:

Create new environment:

conda env create -f ./requirements.yml -n fraud-detection
conda activate fraud-detection

Update existing environment:

conda env update -f ./requirements.yml

3. Start JupyterLab

4. Open and run gnn-fraud-detection-training.ipynb

Relevant log output

Click here to see error details

ImportError                               Traceback (most recent call last)

Cell In[1], line 23

21 from tqdm import trange

22 from xgboost import XGBClassifier

---> 23 from training import (get_metrics, evaluate, init_loaders, build_fsi_graph,

24                      map_node_id, prepare_data, save_model, train)
ImportError: cannot import name 'map_node_id' from 'training' (/workspace/models/training-tuning-scripts/fraud-detection-models/training.py)

Full env printout

No response

Other/Misc.

No response

Code of Conduct

• I agree to follow Morpheus' Code of Conduct
• I have searched the open bugs and have found no duplicates for this bug report