affected_rows fix for gpus

JEMeyer · Jul 29, 2024 · e6cb628 · e6cb628
1 parent 55a3fbf
commit e6cb628
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/services/tables/assignments.ts b/src/services/tables/assignments.ts
@@ -43,15 +43,15 @@ export const getAssignmentById = async (
 export const updateAssignment = async (
   id: number,
   name: string,
-  modelName: string,
+  model_name: string,
   port: number,
   display_order: number
 ): Promise<number> => {
   const query =
     'UPDATE assignments SET name = ?, model_name = ?, port = ?, display_order = ? WHERE id = ?';
   const result = await pool.query(query, [
     name,
-    modelName,
+    model_name,
     port,
     display_order,
     id,

diff --git a/src/services/tables/gpus.ts b/src/services/tables/gpus.ts
@@ -65,12 +65,12 @@ export const updateGPU = async (
     display_order,
     id,
   ]);
-  return Number(result[0].affectedRows);
+  return Number(result.affectedRows);
 };
 
 // Delete a GPU
 export const deleteGPU = async (id: number): Promise<number> => {
   const query = 'DELETE FROM gpus WHERE id = ?';
   const result = await pool.query(query, [id]);
-  return Number(result[0].affectedRows);
+  return Number(result.affectedRows);
 };