Spaces:
Running on Zero
Running on Zero
Make held-out manifests portable
Browse files
prepare_inference_manifest.py
CHANGED
|
@@ -5,6 +5,7 @@ from __future__ import annotations
|
|
| 5 |
import argparse
|
| 6 |
import hashlib
|
| 7 |
import json
|
|
|
|
| 8 |
from pathlib import Path
|
| 9 |
|
| 10 |
from validate_capture_manifest import validate_capture_manifest
|
|
@@ -24,7 +25,7 @@ def prepare_inference_manifest(source_path: str | Path, destination: str | Path)
|
|
| 24 |
prepared = dict(manifest)
|
| 25 |
prepared["views"] = train
|
| 26 |
prepared["evaluation"] = {
|
| 27 |
-
"source_manifest":
|
| 28 |
"source_manifest_sha256": hashlib.sha256(source.read_bytes()).hexdigest(),
|
| 29 |
"excluded_held_out_view_ids": [view["id"] for view in held_out],
|
| 30 |
"held_out_view_count": len(held_out),
|
|
|
|
| 5 |
import argparse
|
| 6 |
import hashlib
|
| 7 |
import json
|
| 8 |
+
import os
|
| 9 |
from pathlib import Path
|
| 10 |
|
| 11 |
from validate_capture_manifest import validate_capture_manifest
|
|
|
|
| 25 |
prepared = dict(manifest)
|
| 26 |
prepared["views"] = train
|
| 27 |
prepared["evaluation"] = {
|
| 28 |
+
"source_manifest": Path(os.path.relpath(source, target.parent)).as_posix(),
|
| 29 |
"source_manifest_sha256": hashlib.sha256(source.read_bytes()).hexdigest(),
|
| 30 |
"excluded_held_out_view_ids": [view["id"] for view in held_out],
|
| 31 |
"held_out_view_count": len(held_out),
|