Python delete_checkpoints Examples

Example #1

File: test_gc_harness.py Project: justin-determined-ai/determined

def test_dry_run(manager: storage.StorageManager, to_delete: List[str]) -> None:
    delete_checkpoints(manager, to_delete, dry_run=True)
    assert len(os.listdir(manager._base_path)) == len(to_delete)

Example #2

File: test_gc_harness.py Project: solversa/determined

def test_validate_failure(config: Dict[str, Any], to_delete: List[Dict[str, Any]]) -> None:
    host_path = config["checkpoint_storage"].pop("host_path")
    with pytest.raises(TypeError):
        delete_checkpoints(config, to_delete, validate=True, dry_run=False)

    assert len(os.listdir(host_path)) == len(to_delete)

Example #3

File: test_gc_harness.py Project: yitang/determined

def test_delete_checkpoints(manager: storage.StorageManager,
                            to_delete: List[Dict[str, Any]]) -> None:
    delete_checkpoints(manager, to_delete, dry_run=False)
    assert len(os.listdir(manager._base_path)) == 0

Example #4

File: test_gc_harness.py Project: solversa/determined

def test_validate_success(config: Dict[str, Any], to_delete: List[Dict[str, Any]]) -> None:
    delete_checkpoints(config, to_delete, validate=True, dry_run=False)

    host_path = config["checkpoint_storage"]["host_path"]
    assert len(os.listdir(host_path)) == 0