GNNs fixed

data_preprocessing/pdlc-dp-deployment.yaml

@@ -38,7 +38,7 @@ spec:
       nodeName: sonem-worker #The node of the Deployment
       containers:
       - name: pdlc-dp
-        image: docker.io/hecodeco/pdlc-dp:v0.2.1
+        image: docker.io/hecodeco/pdlc-dp:v0.2.2
         volumeMounts:
         - name: shared-volume
           mountPath: /data

gnn_model/gnn_inference.yaml

@@ -17,7 +17,7 @@ spec:
       serviceAccountName: default
       containers:
       - name: pdlc-inference
-        image: docker.io/hecodeco/pdlc-dl-gnns-stgnn-inference:1.1.3
+        image: docker.io/hecodeco/pdlc-dl-gnns-stgnn-inference:1.1.4
         # imagePullPolicy: Always
         env:
         - name: NAMESPACE