AttributeError: 'GraphNeuralNetworkModel' object has no attribute '_GraphNeuralNetworkModel__edge_embedding_model'

[Karim-53]

Traceback (most recent call last):
  File "/usr/src/entrypoint.py", line 113, in <module>
    for graph, predictions in model.predict(data_iter(), nn, "cpu"):
  File "/usr/local/lib/python3.8/site-packages/ptgnn/implementations/typilus/graph2class.py", line 276, in predict
    for mb_data, original_datapoints in self.minibatch_iterator(
  File "/usr/local/lib/python3.8/site-packages/ptgnn/baseneuralmodel/abstractneuralmodel.py", line 3[54](https://github.com/Karim-53/Compare-xAI/actions/runs/3900136300/jobs/6661810842#step:4:55), in minibatch_iterator
    yield from ThreadedIterator(
  File "/usr/local/lib/python3.8/site-packages/ptgnn/baseneuralmodel/abstractneuralmodel.py", line 3[55](https://github.com/Karim-53/Compare-xAI/actions/runs/3900136300/jobs/6661810842#step:4:56), in <genexpr>
    ((self.finalize_minibatch(d[0], device), d[1]) for d in unfinalized_minibatches),
  File "/usr/local/lib/python3.8/site-packages/ptgnn/baseneuralmodel/abstractneuralmodel.py", line 301, in __iterate_unfinalized_minibatches
    mb_data = self.initialize_minibatch()
  File "/usr/local/lib/python3.8/site-packages/ptgnn/implementations/typilus/graph2class.py", line 212, in initialize_minibatch
    "graph_mb_data": self.__gnn_model.initialize_minibatch(),
  File "/usr/local/lib/python3.8/site-packages/ptgnn/neuralmodels/gnn/graphneuralnetwork.py", line 374, in initialize_minibatch
    "edge_feature_data": [
  File "/usr/local/lib/python3.8/site-packages/ptgnn/neuralmodels/gnn/graphneuralnetwork.py", line 376, in <listcomp>
    if self.__edge_embedding_model is not None
AttributeError: 'GraphNeuralNetworkModel' object has no attribute '_GraphNeuralNetworkModel__edge_embedding_model'