fix nits

DSGT-DLP · May 14, 2024 · 558fb11 · 558fb11
1 parent a4d8480
commit 558fb11
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/frontend/src/features/Train/redux/trainspaceApi.ts b/frontend/src/features/Train/redux/trainspaceApi.ts
@@ -105,7 +105,7 @@ const trainspaceApi = backendApi
             data_source: trainspaceData.dataSource,
             dataset_data: trainspaceData.datasetData,
             review_data: trainspaceData.reviewData,
-            // model_id: not implemented yet
+            // TODO: add model_id
           },
         }),
       }),

diff --git a/serverless/packages/functions/src/trainspace/create_trainspace.ts b/serverless/packages/functions/src/trainspace/create_trainspace.ts
@@ -47,8 +47,7 @@ export async function handler<APIGatewayProxyHandlerV2>(
           },
         },
         ...(Object.keys(review_data).length && { review_data }),
-        // model_id: { S: eventBody["model_id"] }, # to be implemented
-        // results_s3: { S: eventBody["results_s3"] }, # not needed
+        // model_id: { S: eventBody["model_id"] }, # TODO: add model_id
         status: { S: TrainStatus.QUEUED },
       },
     });