From b3e924674f41f64cc8559e7975505f1557f9c9de Mon Sep 17 00:00:00 2001
From: Gaurav S Deshmukh <deshmukg@gilbreth-fe02.rcac.purdue.edu>
Date: Wed, 20 Sep 2023 11:47:56 -0400
Subject: [PATCH] Add name of struct

---
 src/data.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/data.py b/src/data.py
index 708640b..a09e111 100644
--- a/src/data.py
+++ b/src/data.py
@@ -109,6 +109,9 @@ def process(self, z_cutoffs, node_features, edge_features, max_atoms=12,
                 )
                 data_objects.append(data_obj)
 
+            # Add name of structure
+            data_objects.append(name)
+
             # Save data objects
             torch.save(data_objects, self.processed_path / f"data_{i}.pt")