Fix: Fix project migration script API token parsing

neptune-ai · Aug 28, 2024 · eb7dd50 · eb7dd50
1 parent 4325487
commit eb7dd50
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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: