From bd0f8a88b38cfc02585bdec879b3f17bbf758c99 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 28 Jul 2023 08:17:28 +0200
Subject: [PATCH] [montgomery_shenzhen] Fix use of split filename
---
src/ptbench/data/montgomery_shenzhen/datamodule.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/ptbench/data/montgomery_shenzhen/datamodule.py b/src/ptbench/data/montgomery_shenzhen/datamodule.py
index 0335a2e6..e1173824 100644
--- a/src/ptbench/data/montgomery_shenzhen/datamodule.py
+++ b/src/ptbench/data/montgomery_shenzhen/datamodule.py
@@ -14,9 +14,9 @@ class DataModule(ConcatDataModule):
def __init__(self, split_filename: str):
montgomery_loader = MontgomeryLoader()
- montgomery_split = make_montgomery_split("default.json")
+ montgomery_split = make_montgomery_split(split_filename)
shenzhen_loader = ShenzhenLoader()
- shenzhen_split = make_shenzhen_split("default.json")
+ shenzhen_split = make_shenzhen_split(split_filename)
super().__init__(
splits={
--
GitLab