Chore/add loss column

flockoff/validator/database.py

@@ -39,6 +39,7 @@ def _init_db(self):
             )
             self._add_column_if_not_exists(c, 'miner_scores', 'namespace', 'TEXT')
             self._add_column_if_not_exists(c, 'miner_scores', 'revision', 'TEXT')
+            self._add_column_if_not_exists(c, 'miner_scores', 'raw_loss', 'REAL')
 
             self.conn.commit()
         except sqlite3.Error as e:
@@ -154,6 +155,21 @@ def insert_or_reset_uid(
             logger.error(f"Failed to insert/reset UID {uid}: {str(e)}")
             raise DatabaseError(f"Failed to insert/reset UID: {str(e)}") from e
 
+    def update_raw_loss(self, uid: int, loss: float):
+        """Update the raw loss for a given UID."""
+        try:
+            c = self.conn.cursor()
+            c.execute(
+                "UPDATE miner_scores SET raw_loss = ? WHERE uid = ?", (loss, uid)
+            )
+            if c.rowcount == 0:
+                # If somehow a UID is being updated that wasn't inserted, log a warning or error.
+                logger.warning(f"Attempted to update raw_loss for non-existent UID {uid}, no changes made.")
+            self.conn.commit()
+        except sqlite3.Error as e:
+            logger.error(f"Failed to update raw_loss for UID {uid}: {str(e)}")
+            raise DatabaseError(f"Failed to update raw_loss: {str(e)}") from e
+
     def update_raw_eval_score(self, uid: int, new_raw_score: float):
         """Update the raw evaluation score for a given UID."""
         try:

neurons/validator.py

@@ -501,6 +501,7 @@ async def run_step(self):
 
                     raw_scores_this_epoch[uid] = eval_loss
                     self.score_db.update_raw_eval_score(uid, eval_loss)
+                    self.score_db.update_raw_loss(uid, eval_loss)
                     self.score_db.set_score_revision(uid, ns, revision, self.metagraph.hotkeys[uid])
 
                     bt.logging.info(f"Stored evaluation results for UID {uid}")