WIP getting papermill to run

RWKV · Aug 23, 2023 · 1cb089b · 1cb089b
1 parent a21b493
commit 1cb089b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/notebook/github-runner/github-runner.sh b/notebook/github-runner/github-runner.sh
@@ -112,6 +112,12 @@ mkdir -p $PROJ_DIR/model
 # Run the notebook, and store a copy into the output dir
 # -----
 
+INPUT_FILE_PATH="$PROJ_DIR/datapath/$NOTEBOOK_FILE"
+OUTPUT_FILE_PATH="$PROJ_DIR/output/$NOTEBOOK_FILE"
+mkdir -p "$(dirname "$OUTPUT_FILE_PATH")"
+
 echo "# [NOTE] Running notebook: $NOTEBOOK_FILE"
 cd "$PROJ_DIR"
-papermill "$NOTEBOOK_DIR/$NOTEBOOK_FILE" "$PROJ_DIR/output/$NOTEBOOK_FILE" -k python3 --log-output
+papermill \
+    -k python3 --log-output \
+    "$INPUT_FILE_PATH" "$OUTPUT_FILE_PATH"