From 6c8e54bf6ebb34a8adf84afd7d7a800ac57b867f Mon Sep 17 00:00:00 2001 From: Eli Davis Date: Wed, 7 Sep 2022 10:39:50 -0500 Subject: [PATCH] Added yml support --- opensfm/dataset.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/opensfm/dataset.py b/opensfm/dataset.py index 90aaf9b37..4d4b37263 100644 --- a/opensfm/dataset.py +++ b/opensfm/dataset.py @@ -53,7 +53,10 @@ def __init__(self, data_path: str, io_handler=io.IoFilesystemDefault) -> None: self.load_mask_list() def _config_file(self) -> str: - return os.path.join(self.data_path, "config.yaml") + yaml_path = os.path.join(self.data_path, "config.yaml") + if self.io_handler.isfile(yaml_path): + return yaml_path + return os.path.join(self.data_path, "config.yml") def load_config(self) -> None: config_file_path = self._config_file()