Github actions for unittest

src/grfgnn/gnn.py

@@ -221,10 +221,10 @@ def display_viewable_graph():
     print(ground_truth_indices)
 
     # Create a validation dataloader
-    valLoader: DataLoader = DataLoader(val_set, batch_size=500, shuffle=False)
+    valLoader: DataLoader = DataLoader(val_set, batch_size=100, shuffle=False)
 
     # Setup four graphs
-    fig, axes = plt.subplots(4, figsize=[50, 10])
+    fig, axes = plt.subplots(4, figsize=[20, 10])
     fig.suptitle('Foot Estimated Forces vs. Ground Truth')
 
     # Validate on entire eval_dataset

src/grfgnn/urdfParser.py

@@ -95,7 +95,8 @@ def __init__(self,
                                          parent=connections[0],
                                          child=connections[1])
                     self.edges.append(new_edge)
-                if len(connections) > 2: # TODO: Don't connect children to other children
+                if len(connections
+                       ) > 2:  # TODO: Don't connect children to other children
                     # Setup an edge for each pair of connections
                     count = 0
                     for i in range(0, len(connections)):
@@ -233,10 +234,10 @@ def get_edge_name_to_connections_dict(self):
         it's connected to.
 
         Returns:
-            edge_dict (dict[str, np.array([int, int])]): This dictionary
-                takes the name of an edge in the URDF as input. It 
-                returns an np.array that contains the indices of the 
-                two nodes that this edge is connected to.
+            edge_dict (dict[str, np.array([[int, int],[int, int]])]): 
+                This dictionary takes the name of an edge in the URDF
+                as input. It returns an np.array that contains the 
+                indices of the two nodes that this edge is connected to.
         """
 
         node_dict = self.get_node_name_to_index_dict()