feat(remote-model): metadata update

FocoosAI · Dec 10, 2024 · be4c73a · be4c73a
1 parent 4b57251
commit be4c73a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/focoos/remote_model.py b/focoos/remote_model.py
@@ -39,12 +39,11 @@ def __init__(self, model_ref: str, http_client: HttpClient):
 
     def get_info(self) -> ModelMetadata:
         res = self.http_client.get(f"models/{self.model_ref}")
-        self.metadata = ModelMetadata(**res.json())
-        if res.status_code == 200:
-            return self.metadata
-        else:
+        if res.status_code != 200:
             logger.error(f"Failed to get model info: {res.status_code} {res.text}")
             raise ValueError(f"Failed to get model info: {res.status_code} {res.text}")
+        self.metadata = ModelMetadata(**res.json())
+        return self.metadata
 
     def train(
         self,