Check branch in auto update

Author: HudsonGraeme

neurons/constants.py

@@ -134,3 +134,5 @@ class Roles:
     (57, 237),  # gaia
     (59, 249),  # agent-arena
 ]
+# GitHub repository URL
+REPO_URL = "https://github.com/inference-labs-inc/omron-subnet"

neurons/utils/auto_update.py

@@ -5,6 +5,7 @@
 import sys
 import requests
 from typing import Optional
+from constants import REPO_URL
 
 from bittensor import logging
 
@@ -46,9 +47,7 @@ def get_latest_release_tag(self):
         Get the latest release tag from the GitHub repository
         """
         try:
-            response = requests.get(
-                f"https://api.github.com/repos/inference-labs-inc/omron-subnet/releases/latest"
-            )
+            response = requests.get(f"{REPO_URL}/releases/latest")
             response.raise_for_status()
             latest_release = response.json()
             return latest_release["tag_name"]
@@ -117,11 +116,21 @@ def update_to_latest_release(self) -> bool:
         try:
 
             if self.repo.is_dirty(untracked_files=False):
-                logging.error(
+                logging.warning(
                     "Current changeset is dirty. Please commit changes, discard changes or update manually."
                 )
                 return False
 
+            if self.repo.head.is_detached:
+                logging.warning("Current branch is detached. Please update manually.")
+                return False
+
+            if self.repo.active_branch.name != TARGET_BRANCH:
+                logging.warning(
+                    f"Current branch is not {TARGET_BRANCH}. Please update manually."
+                )
+                return False
+
             latest_release_tag_name = self.get_latest_release_tag()
             if not latest_release_tag_name:
                 logging.error("Failed to fetch the latest release tag.")