Skip to content

Commit

Permalink
Use workaround to fix Ignite test (#526)
Browse files Browse the repository at this point in the history
  • Loading branch information
BryanCutler authored and yongtang committed Oct 2, 2019
1 parent cd94eaa commit bddc061
Showing 1 changed file with 13 additions and 8 deletions.
21 changes: 13 additions & 8 deletions tests/test_ignite.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,6 @@
import tensorflow as tf
tf.compat.v1.disable_eager_execution()

pytest.skip(
"ignite test is disabled temporarily", allow_module_level=True)

from tensorflow import dtypes # pylint: disable=wrong-import-position
from tensorflow import errors # pylint: disable=wrong-import-position
from tensorflow import test # pylint: disable=wrong-import-position
Expand Down Expand Up @@ -307,12 +304,20 @@ def test_ignite_dataset_with_plain_client_with_interleave(self):
"""
self._clear_env()

igds_local = ignite_io.IgniteDataset(
cache_name="SQL_PUBLIC_TEST_CACHE",
schema_host="localhost",
host='localhost',
port=10800)

# TODO: this is a workaround due to failure to build a TypeSpec for
# IgniteDataset in non-eager mode
ds = data.Dataset.from_tensor_slices(["localhost"]).interleave(
lambda host: ignite_io.IgniteDataset(
cache_name="SQL_PUBLIC_TEST_CACHE",
schema_host="localhost", host=host,
port=10800), cycle_length=4, block_length=16
)
lambda host: igds_local,
cycle_length=4,
block_length=16)

self._check_dataset(ds)

def _clear_env(self):
Expand Down

0 comments on commit bddc061

Please sign in to comment.