Skip to content

Commit 44ddac1

Browse files
committed
Fix has_split method
1 parent 9b5d063 commit 44ddac1

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
setup(
99
name="torchsr",
10-
version="1.0.3",
10+
version="1.0.4",
1111
description="Super Resolution Networks for pytorch",
1212
long_description=long_description,
1313
long_description_content_type="text/markdown",

torchsr/datasets/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,15 +110,15 @@ def get_splits(cls):
110110
return set(sp for (t, sp, sc) in cls.track_dirs.keys())
111111

112112
@classmethod
113-
def has_split(self, split):
113+
def has_split(cls, split):
114114
return split in cls.get_splits()
115115

116116
def get_dir(self, track, split, scale):
117117
if (track, split, scale) not in self.track_dirs:
118118
if track not in self.get_tracks():
119119
raise ValueError(f"{self.__class__.__name__} does not include track {track}. "
120120
f"Use one of {list(self.get_tracks())}")
121-
if split not in self.get_splits():
121+
if not self.has_split(split):
122122
raise ValueError(f"{self.__class__.__name__} does not include split {split}. "
123123
f"Use one of {list(self.get_splits())}")
124124
available = ", ".join([str(sc) for t, sp, sc in self.track_dirs if t == track and sp == split])

0 commit comments

Comments
 (0)