fix indexing

Author: camfairchild

pallets/subtensor/src/epoch/math.rs

@@ -612,11 +612,21 @@ pub fn mask_diag_sparse_except_index(
     except_index: u16,
 ) -> Vec<Vec<(u16, I32F32)>> {
     // Store the diagonal entry at except_index
-    let diag_at_index = sparse_matrix[except_index as usize][except_index as usize].clone();
+    let diag_at_index = sparse_matrix
+        .get(except_index as usize)
+        .and_then(|row| row.get(except_index as usize))
+        .cloned();
     // Mask out the diagonal
     let mut result = mask_diag_sparse(sparse_matrix);
-    // Replace the diagonal entry at except_index
-    result[except_index as usize][except_index as usize] = diag_at_index;
+    // Replace the diagonal entry at except_index using only get_mut or map
+    result.get_mut(except_index as usize).map(|row| {
+        row.get_mut(except_index as usize).map(|value| {
+            if let Some(diag_at_index) = diag_at_index {
+                *value = diag_at_index;
+            }
+        })
+    });
+
     result
 }
 

pallets/subtensor/src/utils/misc.rs

@@ -736,11 +736,9 @@ impl<T: Config> Pallet<T> {
 
     // Get the uid of the Owner Hotkey for a subnet.
     pub fn get_owner_uid(netuid: u16) -> Option<u16> {
-        let owner_hotkey = SubnetOwnerHotkey::<T>::get(netuid);
-        if Uids::<T>::contains_key(netuid, &owner_hotkey) {
-            Some(Uids::<T>::get(netuid, &owner_hotkey))
-        } else {
-            None
+        match SubnetOwnerHotkey::<T>::try_get(netuid) {
+            Ok(owner_hotkey) => Uids::<T>::get(netuid, &owner_hotkey),
+            Err(_) => None,
         }
     }
 }