fix: error when Encord is the first downloaded dataset source (#71)

encord-team · May 3, 2024 · 6c45199 · 6c45199
1 parent 2390b5f
commit 6c45199
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tti_eval/dataset/types/encord_ds.py b/tti_eval/dataset/types/encord_ds.py
@@ -130,6 +130,7 @@ def _setup(
  split_to_lr_hashes = json.loads(splits_file.read_text(encoding="utf-8"))
  else:
  split_to_lr_hashes = simple_project_split(self._project)
+ splits_file.parent.mkdir(parents=True, exist_ok=True)
  splits_file.write_text(json.dumps(split_to_lr_hashes), encoding="utf-8")
  self._label_rows = self._project.list_label_rows_v2(label_hashes=split_to_lr_hashes[self.split])