hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 10:01:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Test output length in DatasetReader with default IOContext. (#26852)

Browse source
This commit is contained in:
Rohan Potdar 2022-07-23 04:53:59 -07:00 • committed by GitHub
parent 2cfd6c2e97
commit 69f6b843da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
rllib/offline/tests/test_dataset_reader.py
View file

@ -139,7 +139,8 @@ class TestDatasetReader(unittest.TestCase):
{"input": "dataset", "input_config": input_config} {"input": "dataset", "input_config": input_config}
) )
reader = DatasetReader(dataset) reader = DatasetReader(dataset)
reader.next() # Reads in one line of Pendulum dataset with 600 timesteps
assert len(reader.next()) == 600
class TestUnzipIfNeeded(unittest.TestCase): class TestUnzipIfNeeded(unittest.TestCase):