Pass dataroot to get_prediction_challenge_split() in baseline_model_i…

…nference (#761) Pass dataroot to get_prediction_challenge_split() in baseline_model_inference.
nutonomy · May 9, 2022 · 28765b8 · 28765b8
1 parent 7f99a2a
commit 28765b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-sdk/nuscenes/eval/prediction/baseline_model_inference.py b/python-sdk/nuscenes/eval/prediction/baseline_model_inference.py
@@ -27,7 +27,7 @@ def main(version: str, data_root: str,
 
     nusc = NuScenes(version=version, dataroot=data_root)
     helper = PredictHelper(nusc)
-    dataset = get_prediction_challenge_split(split_name)
+    dataset = get_prediction_challenge_split(split_name, dataroot=data_root)
     config = load_prediction_config(helper, config_name)
     oracle = PhysicsOracle(config.seconds, helper)
     cv_heading = ConstantVelocityHeading(config.seconds, helper)