datachain 0.21.0__py3-none-any.whl → 0.21.1__py3-none-any.whl

datachain/cli/__init__.py

@@ -100,8 +100,10 @@ def handle_command(args, catalog, client_config) -> int:
 
     handler = command_handlers.get(args.command)
     if handler:
-        handler()
-        return 0
+        return_code = handler()
+        if return_code is None:
+            return 0
+        return return_code
     print(f"invalid command: {args.command}", file=sys.stderr)
     return 1
 

datachain/studio.py

@@ -248,14 +248,17 @@ def save_config(hostname, token, level=ConfigLevel.GLOBAL):
 def show_logs_from_client(client, job_id):
     # Sync usage
     async def _run():
+        latest_status = None
         async for message in client.tail_job_logs(job_id):
             if "logs" in message:
                 for log in message["logs"]:
                     print(log["message"], end="")
             elif "job" in message:
-                print(f"\n>>>> Job is now in {message['job']['status']} status.")
+                latest_status = message["job"]["status"]
+                print(f"\n>>>> Job is now in {latest_status} status.")
+        return latest_status
 
-    asyncio.run(_run())
+    latest_status = asyncio.run(_run())
 
     response = client.dataset_job_versions(job_id)
     if not response.ok:
@@ -270,6 +273,12 @@ def show_logs_from_client(client, job_id):
     else:
         print("\n\nNo dataset versions created during the job.")
 
+    exit_code_by_status = {
+        "FAILED": 1,
+        "CANCELLED": 2,
+    }
+    return exit_code_by_status.get(latest_status.upper(), 0) if latest_status else 0
+
 
 def create_job(
     query_file: str,
@@ -326,7 +335,7 @@ def create_job(
     print("Open the job in Studio at", response.data.get("job", {}).get("url"))
     print("=" * 40)
 
-    show_logs_from_client(client, job_id)
+    return show_logs_from_client(client, job_id)
 
 
 def upload_files(client: StudioClient, files: list[str]) -> list[str]:
@@ -396,7 +405,7 @@ def show_job_logs(job_id: str, team_name: Optional[str]):
         )
 
     client = StudioClient(team=team_name)
-    show_logs_from_client(client, job_id)
+    return show_logs_from_client(client, job_id)
 
 
 def list_clusters(team_name: Optional[str]):

{datachain-0.21.0.dist-info → datachain-0.21.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datachain
-Version: 0.21.0
+Version: 0.21.1
 Summary: Wrangle unstructured AI data at scale
 Author-email: Dmitry Petrov <support@dvc.org>
 License-Expression: Apache-2.0

{datachain-0.21.0.dist-info → datachain-0.21.1.dist-info}/RECORD

@@ -15,14 +15,14 @@ datachain/progress.py,sha256=lRzxoYP4Qv2XBwD78sOkmYRzHFpZ2ExVNJF8wAeICtY,770
 datachain/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datachain/script_meta.py,sha256=V-LaFOZG84pD0Zc0NvejYdzwDgzITv6yHvAHggDCnuY,4978
 datachain/semver.py,sha256=UB8GHPBtAP3UJGeiuJoInD7SK-DnB93_Xd1qy_CQ9cU,2074
-datachain/studio.py,sha256=i_i669se7xSjVIChl0aCGyw_DqANUWvTyKegm9QWBrE,12398
+datachain/studio.py,sha256=u74vy3iymPFSXU9ePi7KNOSwv-vQYUUS7BuEoC8cLfs,12697
 datachain/telemetry.py,sha256=0A4IOPPp9VlP5pyW9eBfaTK3YhHGzHl7dQudQjUAx9A,994
 datachain/utils.py,sha256=DNqOi-Ydb7InyWvD9m7_yailxz6-YGpZzh00biQaHNo,15305
 datachain/catalog/__init__.py,sha256=cMZzSz3VoUi-6qXSVaHYN-agxQuAcz2XSqnEPZ55crE,353
 datachain/catalog/catalog.py,sha256=J1MKOuoMSl5B0_XYGF5EjDPm7KCvOvllz8PXxt316Og,59352
 datachain/catalog/datasource.py,sha256=IkGMh0Ttg6Q-9DWfU_H05WUnZepbGa28HYleECi6K7I,1353
 datachain/catalog/loader.py,sha256=UXjYD6BNRoupPvkiz3-b04jepXhtLHCA4gzKFnXxOtQ,5987
-datachain/cli/__init__.py,sha256=eKCyqT05OMESHXCC93iQdqzusBdk1ptqZbBeaEghkgo,8344
+datachain/cli/__init__.py,sha256=AQqMMC4zwBi06XfyDg8mIkCRjL7OtyH7W-9Lq6GyQnc,8421
 datachain/cli/utils.py,sha256=wrLnAh7Wx8O_ojZE8AE4Lxn5WoxHbOj7as8NWlLAA74,3036
 datachain/cli/commands/__init__.py,sha256=zp3bYIioO60x_X04A4-IpZqSYVnpwOa1AdERQaRlIhI,493
 datachain/cli/commands/datasets.py,sha256=77QBkn_Enok0vzkHE0rqCbM9YQuXK1oQNdfCCSKoFKE,5793
@@ -153,9 +153,9 @@ datachain/sql/sqlite/vector.py,sha256=ncW4eu2FlJhrP_CIpsvtkUabZlQdl2D5Lgwy_cbfqR
 datachain/toolkit/__init__.py,sha256=eQ58Q5Yf_Fgv1ZG0IO5dpB4jmP90rk8YxUWmPc1M2Bo,68
 datachain/toolkit/split.py,sha256=ktGWzY4kyzjWyR86dhvzw-Zhl0lVk_LOX3NciTac6qo,2914
 datachain/torch/__init__.py,sha256=gIS74PoEPy4TB3X6vx9nLO0Y3sLJzsA8ckn8pRWihJM,579
-datachain-0.21.0.dist-info/licenses/LICENSE,sha256=8DnqK5yoPI_E50bEg_zsHKZHY2HqPy4rYN338BHQaRA,11344
-datachain-0.21.0.dist-info/METADATA,sha256=BHetpdKogCt4xUdJ2nTjzpPfkEOWK69ILKOkcfy8xR8,13281
-datachain-0.21.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datachain-0.21.0.dist-info/entry_points.txt,sha256=0GMJS6B_KWq0m3VT98vQI2YZodAMkn4uReZ_okga9R4,49
-datachain-0.21.0.dist-info/top_level.txt,sha256=lZPpdU_2jJABLNIg2kvEOBi8PtsYikbN1OdMLHk8bTg,10
-datachain-0.21.0.dist-info/RECORD,,
+datachain-0.21.1.dist-info/licenses/LICENSE,sha256=8DnqK5yoPI_E50bEg_zsHKZHY2HqPy4rYN338BHQaRA,11344
+datachain-0.21.1.dist-info/METADATA,sha256=AgSKtpuTkewsQ0JUUIt_mI3BdzEksWWfE2BpWQuf4HE,13281
+datachain-0.21.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datachain-0.21.1.dist-info/entry_points.txt,sha256=0GMJS6B_KWq0m3VT98vQI2YZodAMkn4uReZ_okga9R4,49
+datachain-0.21.1.dist-info/top_level.txt,sha256=lZPpdU_2jJABLNIg2kvEOBi8PtsYikbN1OdMLHk8bTg,10
+datachain-0.21.1.dist-info/RECORD,,