From a91e61f514ee68c1ab89f0451a06978b18908a6c Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Thu, 26 Oct 2023 12:51:45 +0200
Subject: [PATCH] stupid bug fixed
---
confic.py | 2 +-
datasets.py | 3 +--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/confic.py b/confic.py
index bc446aa..e765147 100644
--- a/confic.py
+++ b/confic.py
@@ -19,6 +19,6 @@ DATA_DIR = 'data/dataset'
OUTDIR = 'model_outputs'
INFERENCE_OUTDIR = 'inference_outputs'
-for required_folders in [TRAIN_DIR, OUTDIR, INFERENCE_OUTDIR]:
+for required_folders in [DATA_DIR, OUTDIR, INFERENCE_OUTDIR]:
if not pathlib.Path(required_folders).exists():
pathlib.Path(required_folders).mkdir(parents=True, exist_ok=True)
diff --git a/datasets.py b/datasets.py
index bcc6270..8fc3020 100644
--- a/datasets.py
+++ b/datasets.py
@@ -64,8 +64,7 @@ def create_train_or_test_dataset(path, train=True):
else:
print('Generate test dataset !')
pfx='test'
- embed()
- quit()
+
csv_candidates = list(Path(path).rglob(f'*{pfx}*.csv'))
if len(csv_candidates) == 0:
print(f'no .csv files for *{pfx}* found in {Path(path)}')