From cf26ffe3d3702a0a77e424c2964ce94e9fe862f8 Mon Sep 17 00:00:00 2001 From: agibsonccc Date: Fri, 26 Mar 2021 13:42:46 +0900 Subject: [PATCH] More test tagging --- .../iterator/EarlyTerminationDataSetIteratorTest.java | 4 ++++ .../iterator/EarlyTerminationMultiDataSetIteratorTest.java | 2 ++ 2 files changed, 6 insertions(+) diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java index 65fff53a3..c0bd12c5b 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationDataSetIteratorTest.java @@ -22,8 +22,10 @@ package org.deeplearning4j.datasets.iterator; import org.deeplearning4j.BaseDL4JTest; import org.deeplearning4j.datasets.iterator.impl.MnistDataSetIterator; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.common.tests.tags.NativeTag; +import org.nd4j.common.tests.tags.TagNames; import org.nd4j.linalg.dataset.DataSet; import org.nd4j.linalg.dataset.api.iterator.DataSetIterator; @@ -37,6 +39,8 @@ import org.junit.jupiter.api.DisplayName; @DisplayName("Early Termination Data Set Iterator Test") @NativeTag +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) class EarlyTerminationDataSetIteratorTest extends BaseDL4JTest { int minibatchSize = 10; diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java index 8413f57f7..a729d0d04 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/datasets/iterator/EarlyTerminationMultiDataSetIteratorTest.java @@ -40,6 +40,8 @@ import static org.junit.jupiter.api.Assertions.*; @DisplayName("Early Termination Multi Data Set Iterator Test") @NativeTag @Tag(TagNames.FILE_IO) +@Tag(TagNames.LARGE_RESOURCES) +@Tag(TagNames.LONG_TEST) class EarlyTerminationMultiDataSetIteratorTest extends BaseDL4JTest { int minibatchSize = 5;