From c1e4613dba3190dfc9057d14375b93efc1514f40 Mon Sep 17 00:00:00 2001 From: "raver119@gmail.com" Date: Fri, 1 May 2020 22:11:56 +0300 Subject: [PATCH] rootDir null validation in ExistingMiniBatchDataSetIterator Signed-off-by: raver119@gmail.com --- .../linalg/dataset/ExistingMiniBatchDataSetIterator.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/ExistingMiniBatchDataSetIterator.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/ExistingMiniBatchDataSetIterator.java index c797fecc0..776b177ba 100644 --- a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/ExistingMiniBatchDataSetIterator.java +++ b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/ExistingMiniBatchDataSetIterator.java @@ -16,6 +16,7 @@ package org.nd4j.linalg.dataset; +import lombok.NonNull; import org.nd4j.linalg.dataset.api.DataSetPreProcessor; import org.nd4j.linalg.dataset.api.iterator.DataSetIterator; @@ -43,7 +44,7 @@ public class ExistingMiniBatchDataSetIterator implements DataSetIterator { * Create with the given root directory, using the default filename pattern {@link #DEFAULT_PATTERN} * @param rootDir the root directory to use */ - public ExistingMiniBatchDataSetIterator(File rootDir) { + public ExistingMiniBatchDataSetIterator(@NonNull File rootDir) { this(rootDir, DEFAULT_PATTERN); } @@ -53,7 +54,7 @@ public class ExistingMiniBatchDataSetIterator implements DataSetIterator { * @param pattern The filename pattern to use. Used with {@code String.format(pattern,idx)}, where idx is an * integer, starting at 0. */ - public ExistingMiniBatchDataSetIterator(File rootDir, String pattern) { + public ExistingMiniBatchDataSetIterator(@NonNull File rootDir, String pattern) { this.rootDir = rootDir; totalBatches = rootDir.list().length; this.pattern = pattern;