diff --git a/predict.py b/predict.py index 3a1a2a0..b226468 100644 --- a/predict.py +++ b/predict.py @@ -21,12 +21,12 @@ def parse_args(): parser = argparse.ArgumentParser() parser.add_argument( "--dataset", - default="v5.2/live.parquet", + default="v5.3/live.parquet", help="Numerapi dataset path or local file.", ) parser.add_argument( "--benchmarks", - default="v5.2/live_benchmark_models.parquet", + default="v5.3/live_benchmark_models.parquet", help="Numerapi benchmark model path or local file.", ) parser.add_argument("--model", required=True, help="Pickled model file or URL") diff --git a/tests/test_predict.py b/tests/test_predict.py index bc474a5..74967e2 100644 --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -80,8 +80,8 @@ def check_inputs(post_url, data, files): def test__main_exits_with_help_on_python_version_pickle_mismatch(self): args = argparse.Namespace( - dataset="v5.2/live.parquet", - benchmarks="v5.2/live_benchmark_models.parquet", + dataset="v5.3/live.parquet", + benchmarks="v5.3/live_benchmark_models.parquet", model="model.pkl", output_dir="/tmp", post_url=None,