mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
Linlang
faa2fb03ad
* fix isort & black & toml-sort & sphinx error * fix ci error * fix ci error * add comments * Update Makefile * change sphinx build command * add auto-lint * add black args * format with black * Auto Linting document * fix ci error --------- Co-authored-by: you-n-g <you-n-g@users.noreply.github.com> Co-authored-by: Young <afe.young@gmail.com>
30 lines
714 B
Python
30 lines
714 B
Python
import fire
|
|
|
|
from rdagent.app.data_mining.conf import PROP_SETTING
|
|
from rdagent.components.workflow.rd_loop import RDLoop
|
|
from rdagent.core.exception import ModelEmptyError
|
|
|
|
|
|
class ModelRDLoop(RDLoop):
|
|
skip_loop_error = (ModelEmptyError,)
|
|
|
|
|
|
def main(path=None, step_n=None):
|
|
"""
|
|
You can continue running session by
|
|
|
|
.. code-block:: python
|
|
|
|
dotenv run -- python rdagent/app/data_mining/model.py $LOG_PATH/__session__/1/0_propose --step_n 1 # `step_n` is a optional paramter
|
|
|
|
"""
|
|
if path is None:
|
|
model_loop = ModelRDLoop(PROP_SETTING)
|
|
else:
|
|
model_loop = ModelRDLoop.load(path)
|
|
model_loop.run(step_n=step_n)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fire.Fire(main)
|