let-else (#116)

open-spaced-repetition · Nov 8, 2023 · 6716304 · 6716304
1 parent f8a1719
commit 6716304
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/src/batch_shuffle.rs b/src/batch_shuffle.rs
@@ -69,11 +69,10 @@ where
     I: Clone + Send + Sync,
 {
     fn get(&self, index: usize) -> Option<I> {
-        if let Some(index) = self.indices.get(index) {
-            self.dataset.get(*index)
-        } else {
-            None
-        }
+        let Some(index) = self.indices.get(index) else {
+            return None;
+        };
+        self.dataset.get(*index)
     }
 
     fn len(&self) -> usize {
@@ -247,11 +246,11 @@ impl<I, O> Iterator for BatchShuffledDataloaderIterator<I, O> {
             }
         }
 
-        if let Some(items) = self.strategy.batch(true) {
-            return Some(self.batcher.batch(items));
-        }
+        let Some(items) = self.strategy.batch(true) else {
+            return None;
+        };
 
-        None
+        Some(self.batcher.batch(items))
     }
 }