test: fix resnet50 CI test
@@ -43,11 +43,7 @@ def test_mnist_tp_to_estimator() -> None:
@pytest.mark.nightly # type: ignore
def test_resnet50() -> None:
- config = conf.load_config(
- conf.experimental_path("resnet50_tf_keras/model_definition/const.yaml")
- )
+ config = conf.load_config(conf.experimental_path("resnet50_tf_keras/const.yaml"))
config = conf.set_max_steps(config, 2)
- exp.run_basic_test_with_temp_config(
- config, conf.experimental_path("resnet50_tf_keras/model_definition"), 1
- )
+ exp.run_basic_test_with_temp_config(config, conf.experimental_path("resnet50_tf_keras"), 1)
