datachain 0.20.0__tar.gz → 0.20.1__tar.gz

{datachain-0.20.0 → datachain-0.20.1}/PKG-INFO

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

{datachain-0.20.0 → datachain-0.20.1}/docs/commands/job/run.md

@@ -24,7 +24,7 @@ This command runs a job in Studio using the specified query file. You can config
 * `--team TEAM` - Team to run job for (default: from config)
 * `--env-file ENV_FILE` - File with environment variables for the job
 * `--env ENV` - Environment variables in KEY=VALUE format
-* `--cluster-id CLUSTER_ID` - Compute cluster ID to run the job on
+* `--cluster CLUSTER` - Compute cluster to run the job on
 * `--workers WORKERS` - Number of workers for the job
 * `--files FILES` - Additional files to include in the job
 * `--python-version PYTHON_VERSION` - Python version for the job (e.g., 3.9, 3.10, 3.11)

{datachain-0.20.0 → datachain-0.20.1}/src/datachain/cli/parser/job.py

@@ -52,10 +52,10 @@ def add_jobs_parser(subparsers, parent_parser) -> None:
     )
 
     studio_run_parser.add_argument(
-        "--cluster-id",
-        type=int,
+        "--cluster",
+        type=str,
         action="store",
-        help="Compute cluster ID to run the job on",
+        help="Compute cluster to run the job on",
     )
 
     studio_run_parser.add_argument(

{datachain-0.20.0 → datachain-0.20.1}/src/datachain/remote/studio.py

@@ -427,7 +427,7 @@ class StudioClient:
         requirements: Optional[str] = None,
         repository: Optional[str] = None,
         priority: Optional[int] = None,
-        cluster_id: Optional[int] = None,
+        cluster: Optional[str] = None,
     ) -> Response[JobData]:
         data = {
             "query": query,
@@ -440,7 +440,7 @@ class StudioClient:
             "requirements": requirements,
             "repository": repository,
             "priority": priority,
-            "compute_cluster_id": cluster_id,
+            "compute_cluster_name": cluster,
         }
         return self._send_request("datachain/job", data)
 

{datachain-0.20.0 → datachain-0.20.1}/src/datachain/studio.py

@@ -41,7 +41,7 @@ def process_jobs_args(args: "Namespace"):
             args.req,
             args.req_file,
             args.priority,
-            args.cluster_id,
+            args.cluster,
         )
 
     if args.cmd == "cancel":
@@ -300,7 +300,7 @@ def create_job(
     req: Optional[list[str]] = None,
     req_file: Optional[str] = None,
     priority: Optional[int] = None,
-    cluster_id: Optional[int] = None,
+    cluster: Optional[str] = None,
 ):
     query_type = "PYTHON" if query_file.endswith(".py") else "SHELL"
     with open(query_file) as f:
@@ -330,7 +330,7 @@ def create_job(
         repository=repository,
         requirements=requirements,
         priority=priority,
-        cluster_id=cluster_id,
+        cluster=cluster,
     )
     if not response.ok:
         raise DataChainError(response.message)

{datachain-0.20.0 → datachain-0.20.1}/src/datachain.egg-info/PKG-INFO

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

{datachain-0.20.0 → datachain-0.20.1}/tests/test_cli_studio.py

@@ -398,8 +398,8 @@ def test_studio_run(capsys, mocker, tmp_dir):
                     "stupidity",
                     "--repository",
                     "https://github.com/iterative/datachain",
-                    "--cluster-id",
-                    "1",
+                    "--cluster",
+                    "default",
                 ]
             )
             == 0
@@ -438,5 +438,5 @@ def test_studio_run(capsys, mocker, tmp_dir):
         "team_name": "team_name",
         "repository": "https://github.com/iterative/datachain",
         "priority": 5,
-        "compute_cluster_id": 1,
+        "compute_cluster_name": "default",
     }