From bc3f1f6e127adc5646eeab045ec9ce6ccea36a20 Mon Sep 17 00:00:00 2001 From: Divyaranjan Sahoo Date: Sun, 5 Oct 2025 13:31:08 +0530 Subject: [PATCH] refactor: use os.path.splitext to strip .txt suffix in training labels --- whereami/get_data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whereami/get_data.py b/whereami/get_data.py index 800bfd4..467d617 100644 --- a/whereami/get_data.py +++ b/whereami/get_data.py @@ -38,5 +38,5 @@ def get_train_data(folder=None): for line in f: data.append(json.loads(line)) X.extend(data) - y.extend([fname.removesuffix(".txt")] * len(data)) + y.extend([os.path.splitext(fname)[0]] * len(data)) return X, y