iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reformulate Math Functions

Browse Source
This commit is contained in:
D-X-Y
2021-04-26 05:16:38 -07:00
parent 1980779053
commit e1818694a4
14 changed files with 308 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_synthetic_env.py
View File

@@ -13,7 +13,7 @@ print("library path: {:}".format(lib_dir))
if str(lib_dir) not in sys.path:
sys.path.insert(0, str(lib_dir))
from datasets import ConstantGenerator, SinGenerator
from datasets import ConstantFunc, ComposedSinFunc
from datasets import SyntheticDEnv
@@ -21,10 +21,10 @@ class TestSynethicEnv(unittest.TestCase):
"""Test the synethtic environment."""
def test_simple(self):
mean_generator = SinGenerator()
std_generator = ConstantGenerator(constant=0.5)
mean_generator = ComposedSinFunc(constant=0.1)
std_generator = ConstantFunc(constant=0.5)
dataset = SyntheticDEnv([mean_generator], [[std_generator]])
dataset = SyntheticDEnv([mean_generator], [[std_generator]], num_per_task=5000)
print(dataset)
for timestamp, tau in dataset:
assert tau.shape == (5000, 1)