def test_report_after_finish(session):
session.start()
session.pause_reporting()
session.finish()
for _ in range(2):
report(loss=1)
assert session.get_next() is None
def test_report():
def train():
for i in range(2):
report(loss=i)
init_session(training_func=train, world_rank=0)
session = get_session()
session.start()
assert session.get_next()["loss"] == 0
assert session.get_next()["loss"] == 1
shutdown_session()
with pytest.raises(ValueError):
report(loss=2)
def train_2():
for i in range(2):
report(loss=i)
train_1()
def train():
for i in range(2):
report(i)
return 1
def train():
for i in range(2):
report(loss=i)
