From 9f75a244e93f3188d02439e09db847f426fca650 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 28 Jul 2023 08:15:59 +0200
Subject: [PATCH] [montgomery_shenzhen] Fix typo in naming
---
src/ptbench/data/montgomery_shenzhen/datamodule.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/ptbench/data/montgomery_shenzhen/datamodule.py b/src/ptbench/data/montgomery_shenzhen/datamodule.py
index 35ba9b9a..0335a2e6 100644
--- a/src/ptbench/data/montgomery_shenzhen/datamodule.py
+++ b/src/ptbench/data/montgomery_shenzhen/datamodule.py
@@ -15,22 +15,22 @@ class DataModule(ConcatDataModule):
def __init__(self, split_filename: str):
montgomery_loader = MontgomeryLoader()
montgomery_split = make_montgomery_split("default.json")
- shenzen_loader = ShenzhenLoader()
- shenzen_split = make_shenzhen_split("default.json")
+ shenzhen_loader = ShenzhenLoader()
+ shenzhen_split = make_shenzhen_split("default.json")
super().__init__(
splits={
"train": [
(montgomery_split["train"], montgomery_loader),
- (shenzen_split["train"], shenzen_loader),
+ (shenzhen_split["train"], shenzhen_loader),
],
"validation": [
(montgomery_split["validation"], montgomery_loader),
- (shenzen_split["validation"], shenzen_loader),
+ (shenzhen_split["validation"], shenzhen_loader),
],
"test": [
(montgomery_split["test"], montgomery_loader),
- (shenzen_split["test"], shenzen_loader),
+ (shenzhen_split["test"], shenzhen_loader),
],
}
)
--
GitLab