release

huggingface · Nov 7, 2024 · e55140b · e55140b
1 parent 74fb15e
commit e55140b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -45,7 +45,7 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 
 logger = Logger().get_logger()
-__version__ = "0.8.26.dev0"
+__version__ = "0.8.26"
 
 
 def is_colab():

diff --git a/src/autotrain/client.py b/src/autotrain/client.py
@@ -4,6 +4,8 @@
 
 import requests
 
+from autotrain import logger
+
 
 AUTOTRAIN_API = os.environ.get("AUTOTRAIN_API", "https://autotrain-projects-autotrain-advanced.hf.space/")
 
@@ -258,7 +260,15 @@ def create(
             "valid_split": valid_split,
         }
         response = requests.post(url, headers=self.headers, json=data)
-        return response.json()
+        if response.status_code == 200:
+            resp = response.json()
+            logger.info(
+                f"Project created successfully. Job ID: {resp['job_id']}. View logs at: https://hf.co/spaces/{resp['job_id']}"
+            )
+            return resp
+        else:
+            logger.error(f"Error creating project: {response.json()}")
+            return response.json()
 
     def get_logs(self, job_id: str):
         url = f"{self.host}/api/logs"