Fix file parsing in __plot_spatial

qitianshi · Jan 10, 2022 · 5fa957b · 5fa957b
1 parent 9df6ac8
commit 5fa957b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/analysis/__main__.py b/analysis/__main__.py
@@ -507,9 +507,11 @@ def __plot_spatial():
 
     plot_files = []
     if QUANTITIES is None:
-        plot_files = [f for f in os.listdir(anl.paths.spatial.root(DATE)) if f.endswith("json")]
+        plot_files = [f for f in os.listdir(anl.paths.spatial.root(DATE))
+            if not f.endswith("json")]
     else:
-        plot_files = [f for f in os.listdir(anl.paths.spatial.root(DATE)) if f in QUANTITIES]
+        plot_files = [f for f in os.listdir(anl.paths.spatial.root(DATE))
+            if not f.endswith("json") and f.strip(".tsv") in QUANTITIES]
 
     # Checks if plot_files is empty.
     if not plot_files: