tracebloc 1.2.63__tar.gz → 1.2.65__tar.gz

{tracebloc-1.2.63 → tracebloc-1.2.65}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.63
+Version: 1.2.65
 Summary: Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace.
 Author-email: tracebloc <lukas@tracebloc.io>
 License-Expression: MIT
@@ -105,7 +105,7 @@ user.login()
 user.upload_model(model_name="my_model")
 
 # 3. Link your model to the dataset — returns a training-plan facade
-training_plan = user.link_model_dataset(dataset_id="<your-dataset-id>")
+training_plan = user.link_model_dataset(dataset_id="<your-dataset-key>")
 
 # 4. Start training
 training_plan.start()

{tracebloc-1.2.63 → tracebloc-1.2.65}/README.md

@@ -44,7 +44,7 @@ user.login()
 user.upload_model(model_name="my_model")
 
 # 3. Link your model to the dataset — returns a training-plan facade
-training_plan = user.link_model_dataset(dataset_id="<your-dataset-id>")
+training_plan = user.link_model_dataset(dataset_id="<your-dataset-key>")
 
 # 4. Start training
 training_plan.start()

{tracebloc-1.2.63 → tracebloc-1.2.65}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tracebloc"
-version = "1.2.63"
+version = "1.2.65"
 description = "Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace."
 license = "MIT"
 license-files = ["license.txt"]

{tracebloc-1.2.63 → tracebloc-1.2.65}/tracebloc/_linking.py

@@ -51,7 +51,6 @@ from .validation.lifelines_checks import (
 )
 from .validation.header_rules import NO_OUTPUT_CLASSES
 from tracebloc.utils.constants import (
-    CLASSLESS_CATEGORIES,
     SKLEARN_FRAMEWORK,
     SCIKIT_SURVIVAL_FRAMEWORK,
     LIFELINES_FRAMEWORK,
@@ -110,19 +109,29 @@ def _status_reason(response):
         return ""
 
 
+#: Where the dashboard shows the key ``link_model_dataset`` takes. The backend's
+#: 403 is the same for an unknown key and a key the caller may not use, on purpose,
+#: so the refusal names the place to copy the right one from.
+DATASET_KEY_HINT = (
+    "The use case's dataset key is in the dashboard: open the use case; on its "
+    'Overview, "On this page" shows it under "Start training".'
+)
+
+
 def _map_link_response(response, dataset_id, load_failed_hint=REUPLOAD_HINT):
     """Map the check-model HTTP response to its parsed JSON body.
 
     Pure: reads only the response object; no I/O, no state. Raises
     :class:`DatasetLinkingError` for every failure shape: 400 (invalid /
-    unknown dataset id; the backend's ``LOAD_FAILED`` for the model gets
+    unknown dataset key; the backend's ``LOAD_FAILED`` for the model gets
     ``load_failed_hint`` instead, the re-upload hint by default), 403 (not
-    permitted — leads with the backend's reason), 409, 204, any other non-200/202 status, an unparseable body,
-    or an ``error`` flag inside the body; returns the parsed body dict
-    otherwise. HTTP-status failures other than the mapped codes chain
-    an :class:`APIError` via ``raise ... from``; a 5xx additionally reads
-    as a transient transport failure and names the HTTP status in the
-    message rather than the generic "Linking Failed!".
+    permitted — leads with the backend's reason, ends with
+    :data:`DATASET_KEY_HINT`), 409, 204, any other non-200/202 status, an
+    unparseable body, or an ``error`` flag inside the body; returns the
+    parsed body dict otherwise. HTTP-status failures other than the mapped
+    codes chain an :class:`APIError` via ``raise ... from``; a 5xx
+    additionally reads as a transient transport failure and names the HTTP
+    status in the message rather than the generic "Linking Failed!".
     """
     if response.status_code == 400:
         body = _parse_api_message(response)
@@ -131,8 +140,8 @@ def _map_link_response(response, dataset_id, load_failed_hint=REUPLOAD_HINT):
             # caller's behind it -- say so, not "no dataset".
             raise DatasetLinkingError(f"{body}\n{load_failed_hint}")
         error_message = (
-            f"Please provide a valid dataset ID.\n"
-            f"There is no dataset with ID: {dataset_id}.\n"
+            f"Please provide a valid dataset key.\n"
+            f"There is no dataset with key {dataset_id}.\n"
         )
         if body:
             error_message += body
@@ -143,15 +152,20 @@ def _map_link_response(response, dataset_id, load_failed_hint=REUPLOAD_HINT):
         # model that can't be used on this dataset (e.g. a non-federatable
         # KNN/GaussianProcess on a federated or third-party-owned dataset,
         # backend #923 follow-up). Lead with the backend's specific reason when
-        # present, but keep the dataset id and a neutral frame around it so a
+        # present, but keep the dataset key and a neutral frame around it so a
         # backend catch-all ("error occurred while linking as ...") or a DRF
         # permission ``detail`` doesn't surface as bare, contextless text.
-        # Fall back to access/ID guidance when the body is empty.
+        # Fall back to access/key guidance when the body is empty. Either way the
+        # last line is DATASET_KEY_HINT.
         body = _parse_api_message(response)
         if body:
-            raise DatasetLinkingError(f"Cannot link dataset ID: {dataset_id}.\n{body}")
+            raise DatasetLinkingError(
+                f"Cannot link the dataset with key {dataset_id}.\n{body}\n"
+                f"{DATASET_KEY_HINT}"
+            )
         raise DatasetLinkingError(
-            f"You are not permitted to link dataset ID: {dataset_id}."
+            f"You are not permitted to link the dataset with key {dataset_id}.\n"
+            f"{DATASET_KEY_HINT}"
         )
 
     if response.status_code == 409:
@@ -321,10 +335,9 @@ def _apply_link_success(
 ):
     """Handle a passed check-model response.
 
-    Prints the success output (byte-identical to the pre-split inline
-    prints), runs the link-time hard-fail checks and the warn-only soft
-    checks, and returns a :class:`LinkSuccessState` with the dataset
-    parameters for the caller to apply. Raises
+    Prints the success output, runs the link-time hard-fail checks and the
+    warn-only soft checks, and returns a :class:`LinkSuccessState` with the
+    dataset parameters for the caller to apply. Raises
     :class:`DatasetLinkingError` when a hard-fail check trips.
     """
     total_images = content["total_images"]
@@ -372,23 +385,10 @@ def _apply_link_success(
 
     text = colored("Assignment successful!", "green")
     print(text, "\n")
-    dataset_params = [
-        f"\n \033[1mDataset Parameters\033[0m\n\n",
-        f"datasetId: {dataset_id}\n",
-        f"totalDatasetSize: {total_dataset_size}\n",
-    ]
-    # The published order the head must follow (backend#5686): ``allClasses``
-    # is a count per class in no index order, so it is only the fallback for a
-    # platform that sends no ``class_order``.
-    class_order = content.get("class_order")
-    order_text = class_order_text(class_order)
-    if order_text is not None:
-        dataset_params.append(f"{order_text}\n")
-    elif category not in CLASSLESS_CATEGORIES:
-        dataset_params.append(f"allClasses: {class_names}\n")
+    # The dataset parameters print once, in the training plan
+    # (``TrainingSummary.get_training_plan``), which ``start()`` prints too.
     if input_contract:  # raw mode's checked input contract (backend#5679)
-        dataset_params.append(f"{input_contract}\n")
-    print(*dataset_params)
+        print(f"{input_contract}\n")
 
     # Hard-fail: for token classification the dataset's label values
     # are BIO tag sequences; the model head must match the count of
@@ -484,6 +484,7 @@ def _apply_link_success(
 
     print("Please set a training plan.")
     resources = content.get("resources")
+    class_order = content.get("class_order")
     return LinkSuccessState(
         total_images=total_images,
         num_classes=num_classes,
@@ -493,5 +494,5 @@ def _apply_link_success(
         table_name=table_name,
         resources=resources if isinstance(resources, dict) else None,
         image_attributes=image_attributes,
-        class_order=class_order if order_text is not None else None,
+        class_order=class_order if class_order_text(class_order) is not None else None,
     )

{tracebloc-1.2.63 → tracebloc-1.2.65}/tracebloc/training/plan.py

@@ -106,17 +106,29 @@ MAX_CYCLES = 100
 MAX_NAME_LENGTH = 200
 
 
-def _print_not_supported(
-    parameter: str, framework: str, message: bool = False
-) -> None:
+def _print_not_supported(parameter: str, framework: str) -> None:
     """Print the "parameter not supported on framework" banner in red."""
-    if message:
-        text = parameter
-    else:
-        text = f"The parameter {parameter} is not supported on {framework}\n"
+    text = f"The parameter {parameter} is not supported on {framework}\n"
     print(colored(text, "red"), "\n")
 
 
+def _print_one_pass(parameter: str, asked: Any, framework: str) -> None:
+    """Say that a one-pass framework keeps ``parameter`` at 1, unless 1 is what
+    was asked: one fit is one pass, and the model's own iteration setting
+    lengthens it."""
+    if type(asked) is int and asked == 1:
+        return
+    print(
+        colored(
+            f"A {framework} model trains in one pass, so {parameter} stays 1. "
+            "To train longer, raise the model's own iteration setting, "
+            "such as max_iter.",
+            "yellow",
+        ),
+        "\n",
+    )
+
+
 class TrainingPlan:
     """Plan-parameter state + validators for :class:`LinkModelDataset`.
 
@@ -147,9 +159,9 @@ class TrainingPlan:
         self._on_invalid = on_invalid
 
         # Plan state. Survival frameworks (sklearn, lifelines, scikit-survival)
-        # train in a single pass — epochs and cycles are both forced to 1;
-        # the `epochs()` / `cycles()` setters below reject any attempt to
-        # override. Non-survival frameworks default to the historical
+        # train in a single pass — epochs and cycles are both forced to 1,
+        # and the `epochs()` / `cycles()` setters below keep them there
+        # (`_print_one_pass`). Non-survival frameworks default to the historical
         # epochs=10, cycles=1 and can be tuned freely.
         is_survival = framework in _SURVIVAL_FRAMEWORKS
         self._epochs: int = 1 if is_survival else 10
@@ -196,11 +208,7 @@ class TrainingPlan:
 
     def epochs(self, epochs: int) -> None:
         if self._framework in _SURVIVAL_FRAMEWORKS:
-            _print_not_supported(
-                f"For {self._framework} framework epochs cannot be updated. Epoch is set to 1",
-                self._framework,
-                message=True,
-            )
+            _print_one_pass("epochs", epochs, self._framework)
             self._epochs = 1
             self._on_valid()
         elif type(epochs) == int and 0 < epochs <= MAX_EPOCHS:
@@ -214,11 +222,7 @@ class TrainingPlan:
 
     def cycles(self, cycles: int) -> None:
         if self._framework in _SURVIVAL_FRAMEWORKS:
-            _print_not_supported(
-                f"For {self._framework} framework cycles cannot be updated. Cycle is set to 1",
-                self._framework,
-                message=True,
-            )
+            _print_one_pass("cycles", cycles, self._framework)
             self._cycles = 1
             self._on_valid()
         elif type(cycles) == int:

{tracebloc-1.2.63 → tracebloc-1.2.65}/tracebloc/training/summary.py

@@ -107,9 +107,9 @@ class TrainingSummary(_FacadeCoordinator):
             # Regression and generative categories have no classes, so the
             # class lines would print a meaningless value (#796).
             has_classes = self._f.category not in CLASSLESS_CATEGORIES
-            # The published order the head must follow, the same block the
-            # link prints (backend#5686); ``allClasses`` (a count per class in
-            # no index order) only when the platform sent no order.
+            # The published order the head must follow (backend#5686);
+            # ``allClasses`` (a count per class in no index order) only when
+            # the platform sent no order.
             order_text = class_order_text(self._f.class_order)
             if order_text is not None:
                 all_classes_line = [f"{order_text}\n\n"]

{tracebloc-1.2.63 → tracebloc-1.2.65}/tracebloc/user.py

@@ -42,7 +42,6 @@ from .utils.general_utils import (
     format_exception,
     get_paths,
     require_login,
-    print_error,
     remove_tmp_file,
     scope_weights_path_to_framework,
 )
@@ -613,7 +612,7 @@ class User:
                 )
 
             if upload:
-                print_error(text=f"\n'{self.__model_name}' upload successful.\n")
+                print(colored(f"\n'{self.__model_name}' upload successful.\n", "green"), "\n")
                 warn_if_seed_skipped(self.__client, self.__model_file_path, self.__weights)
             remove_tmp_file(tmp_dir_path=self.__general_checks_obj.tmp_file_path)
         finally:
@@ -716,7 +715,7 @@ class User:
 
         Raises:
             DatasetLinkingError: No model to link (none uploaded, or both
-                ``model`` and ``template``), the dataset ID is invalid, the
+                ``model`` and ``template``), the dataset key is invalid, the
                 stored model is not one of yours, the model is incompatible
                 with the dataset, or the API returned an unexpected response.
                 Transport/HTTP errors are wrapped; the underlying ``APIError``

{tracebloc-1.2.63 → tracebloc-1.2.65}/tracebloc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.63
+Version: 1.2.65
 Summary: Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace.
 Author-email: tracebloc <lukas@tracebloc.io>
 License-Expression: MIT
@@ -105,7 +105,7 @@ user.login()
 user.upload_model(model_name="my_model")
 
 # 3. Link your model to the dataset — returns a training-plan facade
-training_plan = user.link_model_dataset(dataset_id="<your-dataset-id>")
+training_plan = user.link_model_dataset(dataset_id="<your-dataset-key>")
 
 # 4. Start training
 training_plan.start()