Allow subsequent notebooks to change their directories for relative i…

…mport
tira-io · Nov 25, 2023 · 87c1ee5 · 87c1ee5
1 parent 2f7d470
commit 87c1ee5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tira-ir-starters/pyterrier/run-pyterrier-notebook.py b/tira-ir-starters/pyterrier/run-pyterrier-notebook.py
@@ -18,7 +18,9 @@ def parse_args():
 
 def main(args):
     os.environ['TIRA_INPUT_DIRECTORY'] = os.path.abspath(args.input)
+    os.environ['TIRA_INPUT_DATASET'] = os.path.abspath(args.input)
     os.environ['TIRA_OUTPUT_DIRECTORY'] = os.path.abspath(args.output)
+    os.environ['TIRA_OUTPUT_DIR'] = os.path.abspath(args.output)
 
     if args.chdir:
         print(f'Change directory to allow relative imports to "{args.chdir}".', flush=True)