force path when headless

Author: guilyx

jupyddl/data_analyst.py

@@ -24,15 +24,17 @@ def __get_all_pddl_from_data(self):
         tested_files = []
         domains_problems = []
         i = 0
-        for root, _, files in os.walk("data/", topdown=False):
-            for name in files:
-                if ".gitkeep" in name:
-                    continue
-                tested_files.append(os.getcwd() + "/" + os.path.join(root, name))
-                if i % 2 != 0:
-                    domains_problems.append((tested_files[i - 1], tested_files[i]))
-                i += 1
-        return domains_problems
+        if "DISPLAY" in os.environ:
+            for root, _, files in os.walk("data/", topdown=False):
+                for name in files:
+                    if ".gitkeep" in name:
+                        continue
+                    tested_files.append(os.getcwd() + "/" + os.path.join(root, name))
+                    if i % 2 != 0:
+                        domains_problems.append((tested_files[i - 1], tested_files[i]))
+                    i += 1
+            return domains_problems
+        return [('data/problem.pddl', 'data/domain.pddl')]
 
     def __plot_data(self, times, total_nodes, plot_title):
         plt.plot(total_nodes, times, "b:o")

tests/test_data_analyst.py

@@ -2,7 +2,6 @@
 
 import sys
 from os import path
-
 sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
 from jupyddl.data_analyst import DataAnalyst