Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rootDir null validation in ExistingMiniBatchDataSetIterator 

Signed-off-by: raver119@gmail.com <raver119@gmail.com>
master
raver119@gmail.com 2020-05-01 22:11:56 +03:00
parent e64fbbdc9b
commit c1e4613dba
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/ExistingMiniBatchDataSetIterator.java
View File

@ -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;