From eb7dd50b380910651d1ee3c9d5b33bfca861e3f8 Mon Sep 17 00:00:00 2001
From: Siddhant Sadangi <siddhant.sadangi@gmail.com>
Date: Wed, 28 Aug 2024 13:21:19 +0200
Subject: [PATCH] Fix: Fix project migration script API token parsing

---
 utils/migration_tools/from_another_project/runs_migrator.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/utils/migration_tools/from_another_project/runs_migrator.py b/utils/migration_tools/from_another_project/runs_migrator.py
index 8bc33e1f..73641455 100644
--- a/utils/migration_tools/from_another_project/runs_migrator.py
+++ b/utils/migration_tools/from_another_project/runs_migrator.py
@@ -133,6 +133,7 @@ def exc_handler(exctype, value, tb):
 # %% Get list of runs to be copied
 with neptune.init_project(
     project=SOURCE_PROJECT,
+    api_token=SOURCE_TOKEN,
     mode="read-only",
 ) as neptune_from_project:
     source_runs = neptune_from_project.fetch_runs_table(columns=[]).to_pandas()["sys/id"].values
@@ -298,6 +299,7 @@ def copy_metadata(
 def init_target_run(custom_run_id):
     return neptune.init_run(
         project=TARGET_PROJECT,
+        api_token=TARGET_TOKEN,
         custom_run_id=custom_run_id,
         capture_hardware_metrics=False,
         capture_stderr=False,
@@ -311,6 +313,7 @@ def init_target_run(custom_run_id):
 def copy_run(source_run_id):
     with neptune.init_run(
         project=SOURCE_PROJECT,
+        api_token=SOURCE_TOKEN,
         with_id=source_run_id,
         mode="read-only",
     ) as source_run: