| #!/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__") | |