migrations: fix regression node migrations

Signed-off-by: Ricardo Cañuelo <[email protected]>
JenySadadia · Jan 24, 2024 · 628a6d6 · 628a6d6
1 parent 1a19820
commit 628a6d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/migrations/20231215122000_node_models.py b/migrations/20231215122000_node_models.py
@@ -71,8 +71,8 @@ def upgrade(db: "pymongo.database.Database"):
                 'path': regression.get('path'),
                 'kind': 'regression',
                 'data': {
-                    'pass_node': ObjectId(regression['regression_data'][0]),
-                    'fail_node': ObjectId(regression['regression_data'][1])
+                    'pass_node': ObjectId(regression['regression_data'][0]['_id']),
+                    'fail_node': ObjectId(regression['regression_data'][1]['_id'])
                 },
                 'artifacts': regression.get('artifacts'),
                 'created': regression.get('created'),