small fixes for displaying chart

jaydeluca · Oct 15, 2023 · 9bdb3d3 · 9bdb3d3
1 parent bc73c14
commit 9bdb3d3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -122,15 +122,16 @@ def main(args):
 
     plt.xlabel('Date', fontsize=14)
     plt.ylabel('Count', fontsize=14)
-    plt.title('Test File Count by extension in Instrumentation Directory', fontsize=16)
+    plt.title('Test File Count by Language in Instrumentation Directory', fontsize=16)
     plt.xticks(rotation=45)
 
     plt.legend()
     plt.tight_layout()
 
-    if len(args.output) > 0:
+    if args.output is not None:
         plt.savefig(args.output)
-    plt.draw()
+    else:
+        plt.show()
 
 
 if __name__ == '__main__':

diff --git a/media/example_output.png b/media/example_output.png
diff --git a/media/example_output2.png b/media/example_output2.png