changed path handling to be OS agnostic (#153)

* changed path handling to be OS agnostic * removed unnecessary Path object wrapper Co-authored-by: Marco Kemmerling <[email protected]>
facebookresearch · Jun 10, 2022 · 05caa22 · 05caa22
1 parent de84a6b
commit 05caa22
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mbrl/diagnostics/training_browser.py b/mbrl/diagnostics/training_browser.py
@@ -7,6 +7,7 @@
 import os
 import sys
 from argparse import ArgumentParser
+from pathlib import Path
 
 import pandas as pd
 import yaml
@@ -50,10 +51,9 @@ def reloadExperiments(self, data):
         self.data = []
         for path in data:
             if path.endswith(SOURCE):
+                config_path = Path(path).parent / ".hydra/config.yaml"
                 config = yaml.load(
-                    open(
-                        path.replace("/{}".format(SOURCE), "/.hydra/config.yaml"), "r"
-                    ),
+                    open(config_path, "r"),
                     Loader=yaml.FullLoader,
                 )
             elif path.endswith(MULTI_ROOT):