Merge pull request #40 from mit-acl/fix/no-gt-result-viz

Fix title setting for associations plot

Author: liqyn

roman/align/results.py

@@ -62,7 +62,7 @@ def plot_align_results(results: SubmapAlignResults, dpi=500):
         ax = np.array(ax).reshape(-1, 1)
         mp = ax[0, 0].imshow(results.clipper_num_associations, cmap='viridis', vmin=0)
         fig.colorbar(mp, fraction=0.04, pad=0.04)
-        ax.set_title("Number of Associations")
+        ax[0, 0].set_title("Number of Associations")
 
         if show_sim:
             mp = ax[1, 0].imshow(results.similarity_mat, cmap='viridis', vmin=0.0, vmax=1.0)
@@ -270,4 +270,4 @@ def submaps_from_align_results(results: SubmapAlignResults, gt_paths: Tuple[str,
 
     submaps = [submaps_from_roman_map(
         roman_maps[i], submap_params, gt_pose_data[i]) for i in range(2)]
-    return submaps
+    return submaps