Skip to content

Commit 5f24766

Browse files
committed
Removed todo
1 parent 09dc8df commit 5f24766

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

wildlife_tools/features/deep.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -120,9 +120,6 @@ def __call__(self, dataset: ImageDataset) -> FeatureDataset:
120120
self.model = self.model.to(self.device)
121121
self.model = self.model.eval()
122122

123-
# TODO: this is hacky
124-
dataset.transforms = None # Reset transforms.
125-
126123
check_dataset_output(dataset, check_label=False)
127124
loader = torch.utils.data.DataLoader(
128125
dataset,

0 commit comments

Comments
 (0)