예제 #1
0
def test_convert_weight_from_darknet_configs():
    """Test that all config files in `configs/convert_weight_from_darknet` dir include requirement keys."""
    dir_path = os.path.join("configs", "convert_weight_from_darknet")

    for config_file in glob.glob(os.path.join(dir_path, "**", "*.py"),
                                 recursive=True):
        config = _load_py(config_file)
        check_config(config, "inference")
예제 #2
0
def test_core_configs():
    """Test that all config files in `configs/core` dir include requirement keys."""
    dir_path = os.path.join("configs", "core")

    for config_file in glob.glob(os.path.join(dir_path, "**", "*.py"),
                                 recursive=True):
        config = _load_py(config_file)
        check_config(config, "training")
        check_config(config, "inference")
예제 #3
0
def test_example_config():
    """Test that example config python file include requirement keys."""

    dir_path = os.path.join("configs", "example")

    for config_file in glob.glob(os.path.join(dir_path, "**", "*.py"),
                                 recursive=True):
        config = _load_py(config_file)
        check_config(config, "training")
        check_config(config, "inference")