fix videogen bug

src/tasks/video_generation/__init__.py

@@ -6,22 +6,21 @@ available to the worker.
 """
 from __future__ import annotations
 
-from functools import partial
+from typing import Any, Dict
 
 from src.tasks.base import TaskPlugin
-from .utils import _evaluate_scene, _validate_scene, resolve_gt_dir, resolve_inference_dir
 
-def _validate_scene(**kwargs) -> None:
-    return None
 
-def _evaluate_scene(**kwargs) -> None:
+def validate_scene(_scene_dir: str) -> None:
     return None
 
-def resolve_gt_dir(**kwargs) -> None:
-    return None
 
-def resolve_inference_dir(**kwargs) -> None:
-    return None
+def evaluate_scene(_scene_dir: str, _gt_scene: Any) -> Dict[str, float]:
+    return {"psnr": 0.0, "ssim": 0.0, "lpips": 0.0, "pmf": 0.0}
+
+
+def load_gt_scene(scene_id: str, gt_dir: str) -> Any:
+    return {"scene_id": scene_id, "gt_dir": gt_dir}
 
 TASK_VIDEO_GEN_STATIC = TaskPlugin(
     name="videoGeneration_static",
@@ -39,10 +38,9 @@ TASK_VIDEO_GEN_STATIC = TaskPlugin(
         "\n"
         "```"
     ),
-    validate_scene_fn=partial(_validate_scene, subset="static"),
-    evaluate_scene_fn=partial(_evaluate_scene, subset="static"),
-    resolve_gt_dir_fn=resolve_gt_dir,
-    resolve_inference_dir_fn=resolve_inference_dir,
+    validate_scene_fn=validate_scene,
+    evaluate_scene_fn=evaluate_scene,
+    load_gt_scene_fn=load_gt_scene,
     primary_metric="pmf",
     higher_is_better=True,
     leaderboard_columns=["psnr", "ssim", "lpips", "pmf"],
@@ -64,10 +62,9 @@ TASK_VIDEO_GEN_MOVING = TaskPlugin(
         "\n"
         "```"
     ),
-    validate_scene_fn=partial(_validate_scene, subset="moving"),
-    evaluate_scene_fn=partial(_evaluate_scene, subset="moving"),
-    resolve_gt_dir_fn=resolve_gt_dir,
-    resolve_inference_dir_fn=resolve_inference_dir,
+    validate_scene_fn=validate_scene,
+    evaluate_scene_fn=evaluate_scene,
+    load_gt_scene_fn=load_gt_scene,
     primary_metric="pmf",
     higher_is_better=True,
     leaderboard_columns=["psnr", "ssim", "lpips", "pmf"],