tracebloc 1.2.44__tar.gz → 1.2.45__tar.gz

{tracebloc-1.2.44 → tracebloc-1.2.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.44
+Version: 1.2.45
 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

{tracebloc-1.2.44 → tracebloc-1.2.45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tracebloc"
-version = "1.2.44"
+version = "1.2.45"
 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.44 → tracebloc-1.2.45}/tracebloc/_linking.py

@@ -38,6 +38,7 @@ from .validation.lifelines_checks import (
     non_averageable_lifelines_estimator_reason,
 )
 from tracebloc.utils.constants import (
+    CLASSLESS_CATEGORIES,
     SKLEARN_FRAMEWORK,
     SCIKIT_SURVIVAL_FRAMEWORK,
     LIFELINES_FRAMEWORK,
@@ -48,10 +49,6 @@ from tracebloc.utils.constants import (
     TABULAR_CLASSIFICATION,
     TABULAR_REGRESSION,
     KEYPOINT_DETECTION,
-    MASKED_LANGUAGE_MODELING,
-    CAUSAL_LANGUAGE_MODELING,
-    SEQ2SEQ,
-    EMBEDDINGS,
     TIME_SERIES_FORECASTING,
     TIME_SERIES_CLASSIFICATION,
 )
@@ -220,14 +217,7 @@ def _render_mismatch_diagnostics(
     ``print``'s implicit newline).
     """
     expected_msg = ""
-    if category not in [
-        TABULAR_REGRESSION,
-        TIME_SERIES_FORECASTING,
-        MASKED_LANGUAGE_MODELING,
-        CAUSAL_LANGUAGE_MODELING,
-        SEQ2SEQ,
-        EMBEDDINGS,
-    ]:
+    if category not in CLASSLESS_CATEGORIES:
         expected_msg += f"  output_classes : {content.get('datasetClasses')}\n"
     if "datasetShape" in content:
         expected_msg += f"  shape          : {content['datasetShape']}\n"
@@ -254,14 +244,7 @@ def _render_mismatch_diagnostics(
         expected_msg = f"  {content}\n"
 
     model_msg = ""
-    if category not in [
-        TABULAR_REGRESSION,
-        TIME_SERIES_FORECASTING,
-        MASKED_LANGUAGE_MODELING,
-        CAUSAL_LANGUAGE_MODELING,
-        SEQ2SEQ,
-        EMBEDDINGS,
-    ]:
+    if category not in CLASSLESS_CATEGORIES:
         model_msg += f"  output_classes : {content.get('outputClass')}\n"
     if "inputShape" in content:
         model_msg += f"  shape          : {content['inputShape']}\n"
@@ -366,14 +349,7 @@ def _apply_link_success(
         f"datasetId: {dataset_id}\n",
         f"totalDatasetSize: {total_dataset_size}\n",
     ]
-    if category not in [
-        TABULAR_REGRESSION,
-        TIME_SERIES_FORECASTING,
-        MASKED_LANGUAGE_MODELING,
-        CAUSAL_LANGUAGE_MODELING,
-        SEQ2SEQ,
-        EMBEDDINGS,
-    ]:
+    if category not in CLASSLESS_CATEGORIES:
         dataset_params.append(f"allClasses: {class_names}\n")
     print(*dataset_params)
 

{tracebloc-1.2.44 → tracebloc-1.2.45}/tracebloc/training/summary.py

@@ -25,6 +25,7 @@ from tracebloc.training.shape import (
 )
 from tracebloc.training.submitter import _FacadeCoordinator
 from tracebloc.utils.constants import (
+    CLASSLESS_CATEGORIES,
     TABULAR_CLASSIFICATION,
     TIME_SERIES_FORECASTING,
     TIME_SERIES_CLASSIFICATION,
@@ -102,6 +103,17 @@ class TrainingSummary(_FacadeCoordinator):
                 if self._is_image_category()
                 else []
             )
+            # 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
+            all_classes_line = (
+                [f"allClasses: {self._f.class_names}\n\n"] if has_classes else []
+            )
+            training_classes_line = (
+                [f"trainingClasses: {self._f.training_classes}\n"]
+                if has_classes
+                else []
+            )
             print(
                 f" \033[1mTraining Description\033[0m\n\n",
                 f"experimentName: {self._f.plan._name}\n",
@@ -110,9 +122,9 @@ class TrainingSummary(_FacadeCoordinator):
                 f"\n \033[1mDataset Parameters\033[0m\n\n",
                 f"datasetId: {self._f.dataset_id}\n",
                 f"totalDatasetSize: {self._f.total_dataset_size}\n",
-                f"allClasses: {self._f.class_names}\n\n",
+                *all_classes_line,
                 f"trainingDatasetSize: {self._f.training_dataset_size}\n",
-                f"trainingClasses: {self._f.training_classes}\n",
+                *training_classes_line,
                 f"{shape_label}: {shape_value}\n",
                 *data_type_line,
                 f"seed: {self._f.plan._seed}\n",

{tracebloc-1.2.44 → tracebloc-1.2.45}/tracebloc/upload/model_upload.py

@@ -70,6 +70,7 @@ from tracebloc.utils.constants import (
     PRETRAINED_WEIGHTS_FILENAME,
     TRAINED_WEIGHTS_FILENAME,
     AVERAGED_WEIGHTS_PATH,
+    missing_loss_message,
 )
 
 # Timeout for the /check-model/ upload call. The server runs a
@@ -840,9 +841,7 @@ class Model:
         if custom_loss or (category == OBJECT_DETECTION and model_type == YOLO):
             loss_file_path = os.path.join(self.tmp_dir_path, "loss.py")
             if not os.path.exists(loss_file_path):
-                raise FileNotFoundError(
-                    "loss.py file missing in the zip.\nPlease refer to the documentation for more information."
-                )
+                raise FileNotFoundError(missing_loss_message(category, model_type))
 
             # Load loss.py FRESH by explicit file path every call. Using the
             # global ``loss`` module name (``from loss import Custom_loss``)

{tracebloc-1.2.44 → tracebloc-1.2.45}/tracebloc/utils/constants.py

@@ -106,6 +106,21 @@ EMBEDDINGS = TaskCategory.EMBEDDINGS
 TOKEN_CLASSIFICATION = TaskCategory.TOKEN_CLASSIFICATION
 SENTENCE_PAIR_CLASSIFICATION = TaskCategory.SENTENCE_PAIR_CLASSIFICATION
 
+# Categories whose datasets carry no classes: regression targets
+# (tabular regression, time-series forecasting) and the generative /
+# representation text tasks. Every surface that prints a dataset's
+# classes (``allClasses`` / ``trainingClasses`` / ``output_classes``)
+# leaves them out for these. A tuple, not a set: ``in`` then compares by
+# equality, so a plain category string matches its enum member.
+CLASSLESS_CATEGORIES = (
+    TABULAR_REGRESSION,
+    TIME_SERIES_FORECASTING,
+    MASKED_LANGUAGE_MODELING,
+    CAUSAL_LANGUAGE_MODELING,
+    SEQ2SEQ,
+    EMBEDDINGS,
+)
+
 
 # Platform-canonical time-series defaults.
 # The backend Experiment model is the single source of truth for a no-touch
@@ -190,3 +205,28 @@ SUCCESS = "success"
 TORCH_HUB_PATTERN = "torch.hub"
 DNN = "fcn"
 ML = "ml"
+
+
+def missing_loss_message(category=None, model_type=None) -> str:
+    """The error ``configure_loss`` raises when the upload carries no ``loss.py``.
+
+    A YOLO object detector always trains with its own ``loss.py``, so the
+    model zoo ships each YOLO template as a folder (``model.py`` + ``loss.py``)
+    next to a ready-to-upload flat zip of it. Uploading the folder's
+    ``model.py`` alone lands here, so the YOLO message names the zip.
+    """
+    if category == OBJECT_DETECTION and model_type == YOLO:
+        return (
+            "loss.py file missing in the zip. A YOLO model trains with its "
+            "own loss.py, so upload model.py and loss.py together as one zip.\n"
+            "For a model zoo YOLO template, upload the ready-made zip next to "
+            "its folder, e.g. model_zoo/object_detection/pytorch/yolo_v8.zip "
+            "instead of yolo_v8/model.py.\n"
+            "For your own YOLO model, zip model.py and loss.py flat (both at "
+            "the root of the zip, no folder inside) and upload the .zip.\n"
+            "Please refer to the documentation for more information."
+        )
+    return (
+        "loss.py file missing in the zip.\n"
+        "Please refer to the documentation for more information."
+    )

{tracebloc-1.2.44 → tracebloc-1.2.45}/tracebloc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.44
+Version: 1.2.45
 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