File size: 1,079 Bytes
38a51ff | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | #!/usr/bin/env python
"""Run HY-WorldPlay-DEV-Predictor's generate_predictor_v4_train_case_eval.py with the
DisCa predictor: the current DEV rollout passes ``anchor_distance`` (an ATC input) to
every predictor, which HYWorldPlayPredictorDisCa.forward does not accept. Wrap its
forward to drop every keyword it does not declare; everything else is the stock DEV script."""
import os, runpy, sys
DEV = "/local/zoubin/cz/projects/HY-WorldPlay-DEV-Predictor"
sys.path.insert(0, DEV); os.chdir(DEV)
import models # noqa: E402
_orig = models.HYWorldPlayPredictorDisCa.forward
import inspect # noqa: E402
_accepted = set(inspect.signature(_orig).parameters)
def forward(self, *args, **kw):
# drop inputs the ATC-era rollout passes that DisCa never took
# (anchor_distance, previous_step_chunk_hidden, ...)
return _orig(self, *args, **{k: v for k, v in kw.items() if k in _accepted})
models.HYWorldPlayPredictorDisCa.forward = forward
sys.argv[0] = os.path.join(DEV, "tools/generate_predictor_v4_train_case_eval.py")
runpy.run_path(sys.argv[0], run_name="__main__")
|