BenchmarkRunner: Create parent dirs in save_results

This commit is contained in:
2021-01-14 21:00:52 -06:00
parent 622d132ba2
commit fab7b5419b

View File

@@ -8,6 +8,7 @@ import pandas as pd
import numpy as np import numpy as np
import logging import logging
from tqdm.auto import tqdm from tqdm.auto import tqdm
import os
from .solvers.learning import LearningSolver from .solvers.learning import LearningSolver
@@ -61,6 +62,7 @@ class BenchmarkRunner:
return self.results return self.results
def save_results(self, filename): def save_results(self, filename):
os.makedirs(os.path.dirname(filename), exist_ok=True)
self.results.to_csv(filename) self.results.to_csv(filename)
def load_results(self, filename): def load_results(self, filename):