def instantiate_system():
system = sample_system_object()
x_train = np.ones((2, 28, 28, 3))
y_train = np.ones((2, ))
data = {
0: {
'x': x_train[0],
'y': y_train[0]
},
1: {
'x': x_train[1],
'y': y_train[1]
}
}
train_data = RepeatScheduler([
TestNonTraceableDataset(data=data, var=3),
TestDataset(data={
'x': x_train,
'y': y_train
}, var=7), None
])
system.pipeline = fe.Pipeline(train_data=train_data, batch_size=1)
return system
def setUpClass(cls):
cls.scheduler = RepeatScheduler(['a', 'b', 'c', 'c'])
cls.epoch_scheduler = EpochScheduler({1: 'a', 2: 'b', 30: None})
cls.actual_current_items = ['a', 'b', 'c', 'c']
cls.signature_epochs = [1, 2, 3, 5, 30, 31, 33]
def instantiate_system():
system = sample_system_object_torch()
system.traces.append(
RepeatScheduler([TestTrace(var1=1),
TestTrace(var1=1)]))
return system
def setUpClass(cls):
cls.input_data = [1, 2, 2]
cls.scheduler = RepeatScheduler(cls.input_data)
