refactor: refactor tests to go in the unified path of test mode
@@ -5,9 +5,9 @@ from tests.experiment.fixtures import pytorch_xor_model
def test_test_one_batch() -> None:
+ with experimental._local_execution_manager(pathlib.Path(pytorch_xor_model.__file__).parent):
experimental.test_one_batch(
- pathlib.Path(pytorch_xor_model.__file__).parent,
- pytorch_xor_model.XORTrial,
+ trial_class=pytorch_xor_model.XORTrial,
config={
"hyperparameters": {"hidden_size": 2, "learning_rate": 0.5, "global_batch_size": 4}
},
