truss 0.10.11rc2__py3-none-any.whl → 0.10.12__py3-none-any.whl

truss/cli/train/common.py

@@ -5,6 +5,12 @@ import rich_click as click
 from truss.cli.utils.output import console
 from truss.remote.baseten import BasetenRemote
 
+# Byte size constants
+BYTES_PER_KB = 1000
+BYTES_PER_MB = BYTES_PER_KB * 1000
+BYTES_PER_GB = BYTES_PER_MB * 1000
+BYTES_PER_TB = BYTES_PER_GB * 1000
+
 
 def get_most_recent_job(
     remote_provider: BasetenRemote, project_id: Optional[str], job_id: Optional[str]
@@ -27,3 +33,16 @@ def get_most_recent_job(
         project_id = cast(str, job["training_project"]["id"])
         job_id = cast(str, job["id"])
     return project_id, job_id
+
+
+def format_bytes_to_human_readable(bytes: int) -> str:
+    if bytes > BYTES_PER_TB:
+        return f"{bytes / BYTES_PER_TB:.2f} TB"
+    if bytes > BYTES_PER_GB:
+        return f"{bytes / BYTES_PER_GB:.2f} GB"
+    elif bytes > BYTES_PER_MB:
+        return f"{bytes / BYTES_PER_MB:.2f} MB"
+    elif bytes > BYTES_PER_KB:
+        return f"{bytes / BYTES_PER_KB:.2f} KB"
+    else:
+        return f"{bytes} B"

truss/cli/train/core.py

@@ -20,6 +20,14 @@ from truss.remote.baseten.remote import BasetenRemote
 from truss_train import loader
 from truss_train.definitions import DeployCheckpointsConfig
 
+SORT_BY_FILEPATH = "filepath"
+SORT_BY_SIZE = "size"
+SORT_BY_MODIFIED = "modified"
+SORT_BY_TYPE = "type"
+SORT_BY_PERMISSIONS = "permissions"
+SORT_ORDER_ASC = "asc"
+SORT_ORDER_DESC = "desc"
+
 ACTIVE_JOB_STATUSES = [
     "TRAINING_JOB_RUNNING",
     "TRAINING_JOB_CREATED",
@@ -401,3 +409,131 @@ def download_checkpoint_artifacts(
 
 def status_page_url(remote_url: str, training_job_id: str) -> str:
     return f"{remote_url}/training/jobs/{training_job_id}"
+
+
+def fetch_project_by_name_or_id(
+    remote_provider: BasetenRemote, project_identifier: str
+) -> dict:
+    """Fetch a training project by name or ID.
+
+    Args:
+        remote_provider: The remote provider instance
+        project_identifier: Either a project ID or project name
+
+    Returns:
+        The project object as a dictionary
+
+    Raises:
+        click.ClickException: If the project is not found
+    """
+    try:
+        projects = remote_provider.api.list_training_projects()
+        projects_by_name = {project.get("name"): project for project in projects}
+        projects_by_id = {project.get("id"): project for project in projects}
+        if project_identifier in projects_by_id:
+            return projects_by_id[project_identifier]
+        if project_identifier in projects_by_name:
+            return projects_by_name[project_identifier]
+        valid_project_ids_and_names = ", ".join(
+            [f"{project.get('id')} ({project.get('name')})" for project in projects]
+        )
+        raise click.ClickException(
+            f"Project '{project_identifier}' not found. Valid project IDs and names: {valid_project_ids_and_names}"
+        )
+    except click.ClickException:
+        raise
+    except Exception as e:
+        raise click.ClickException(f"Error fetching project: {str(e)}")
+
+
+def view_cache_summary(
+    remote_provider: BasetenRemote,
+    project_id: str,
+    sort_by: str = SORT_BY_FILEPATH,
+    order: str = SORT_ORDER_ASC,
+):
+    """View cache summary for a training project."""
+    try:
+        cache_data = remote_provider.api.get_cache_summary(project_id)
+
+        if not cache_data:
+            console.print("No cache summary found for this project.", style="yellow")
+            return
+
+        table = rich.table.Table(title=f"Cache summary for project: {project_id}")
+        table.add_column("File Path", style="cyan")
+        table.add_column("Size", style="green")
+        table.add_column("Modified", style="yellow")
+        table.add_column("Type")
+        table.add_column("Permissions", style="magenta")
+
+        files = cache_data.get("file_summaries", [])
+        if not files:
+            console.print("No files found in cache.", style="yellow")
+            return
+
+        reverse = order == SORT_ORDER_DESC
+
+        if sort_by == SORT_BY_FILEPATH:
+            files.sort(key=lambda x: x.get("path", ""), reverse=reverse)
+        elif sort_by == SORT_BY_SIZE:
+            files.sort(key=lambda x: x.get("size_bytes", 0), reverse=reverse)
+        elif sort_by == SORT_BY_MODIFIED:
+            files.sort(key=lambda x: x.get("modified", ""), reverse=reverse)
+        elif sort_by == SORT_BY_TYPE:
+            files.sort(key=lambda x: x.get("file_type", ""), reverse=reverse)
+        elif sort_by == SORT_BY_PERMISSIONS:
+            files.sort(key=lambda x: x.get("permissions", ""), reverse=reverse)
+
+        total_size = 0
+        for file_info in files:
+            total_size += file_info.get("size_bytes", 0)
+
+        total_size_str = common.format_bytes_to_human_readable(total_size)
+
+        console.print(
+            f"📅 Cache captured at: {cache_data.get('timestamp', 'Unknown')}",
+            style="bold blue",
+        )
+        console.print(
+            f"📁 Project ID: {cache_data.get('project_id', 'Unknown')}",
+            style="bold blue",
+        )
+        console.print()
+        console.print(f"📊 Total files: {len(files)}", style="bold green")
+        console.print(f"💾 Total size: {total_size_str}", style="bold green")
+        console.print()
+
+        for file_info in files:
+            size_bytes = file_info.get("size_bytes", 0)
+
+            size_str = cli_common.format_bytes_to_human_readable(int(size_bytes))
+
+            modified_str = cli_common.format_localized_time(
+                file_info.get("modified", "Unknown")
+            )
+
+            table.add_row(
+                file_info.get("path", "Unknown"),
+                size_str,
+                modified_str,
+                file_info.get("file_type", "Unknown"),
+                file_info.get("permissions", "Unknown"),
+            )
+
+        console.print(table)
+
+    except Exception as e:
+        console.print(f"Error fetching cache summary: {str(e)}", style="red")
+        raise
+
+
+def view_cache_summary_by_project(
+    remote_provider: BasetenRemote,
+    project_identifier: str,
+    sort_by: str = SORT_BY_FILEPATH,
+    order: str = SORT_ORDER_ASC,
+):
+    """View cache summary for a training project by ID or name."""
+    project = fetch_project_by_name_or_id(remote_provider, project_identifier)
+    view_cache_summary(remote_provider, project["id"], sort_by, order)

truss/cli/train/metrics_watcher.py

@@ -35,18 +35,17 @@ class MetricsWatcher(TrainingPollerMixin):
 
     def _format_bytes(self, bytes_val: float) -> Tuple[str, str]:
         """Convert bytes to human readable format"""
+        default_color = "green"
         color_map = {"MB": "green", "GB": "cyan", "TB": "magenta"}
-        unit = "MB"
-        if bytes_val > 1024 * 1024 * 1024 * 1024:
+        unit = "B"
+        if bytes_val > 1000 * 1000 * 1000 * 1000:
             unit = "TB"
-        elif bytes_val > 1024 * 1024 * 1024:
+        elif bytes_val > 1000 * 1000 * 1000:
             unit = "GB"
-
-        if unit == "MB":
-            return f"{bytes_val / (1024 * 1024):.2f} MB", color_map[unit]
-        elif unit == "GB":
-            return f"{bytes_val / (1024 * 1024 * 1024):.2f} GB", color_map[unit]
-        return f"{bytes_val:.2f} bytes", color_map[unit]
+        elif bytes_val > 1000 * 1000:
+            unit = "MB"
+        color = color_map.get(unit, default_color)
+        return (common.format_bytes_to_human_readable(int(bytes_val)), color)
 
     def _format_storage_utilization(self, utilization: float) -> Tuple[str, str]:
         percent = round(utilization * 100, 4)

truss/cli/train_commands.py

@@ -11,11 +11,21 @@ from truss.cli.logs import utils as cli_log_utils
 from truss.cli.logs.training_log_watcher import TrainingLogWatcher
 from truss.cli.train import common as train_common
 from truss.cli.train import core
+from truss.cli.train.core import (
+    SORT_BY_FILEPATH,
+    SORT_BY_MODIFIED,
+    SORT_BY_PERMISSIONS,
+    SORT_BY_SIZE,
+    SORT_BY_TYPE,
+    SORT_ORDER_ASC,
+    SORT_ORDER_DESC,
+)
 from truss.cli.utils import common
 from truss.cli.utils.output import console, error_console
 from truss.remote.baseten.core import get_training_job_logs_with_pagination
 from truss.remote.baseten.remote import BasetenRemote
 from truss.remote.remote_factory import RemoteFactory
+from truss_train import TrainingJob
 
 
 @click.group()
@@ -27,15 +37,30 @@ truss_cli.add_command(train)
 
 
 def _print_training_job_success_message(
-    job_id: str, remote_provider: BasetenRemote
+    job_id: str,
+    project_name: str,
+    job_object: TrainingJob,
+    remote_provider: BasetenRemote,
 ) -> None:
     """Print success message and helpful commands for a training job."""
     console.print("✨ Training job successfully created!", style="green")
+    should_print_cache_summary = (
+        job_object.runtime.enable_cache
+        or job_object.runtime.cache_config
+        and job_object.runtime.cache_config.enabled
+    )
+    cache_summary_snippet = ""
+    if should_print_cache_summary:
+        cache_summary_snippet = (
+            f"📁 View cache summary via "
+            f"[cyan]'truss train cache summarize --project {project_name}'[/cyan]\n"
+        )
     console.print(
         f"🪵 View logs for your job via "
         f"[cyan]'truss train logs --job-id {job_id} --tail'[/cyan]\n"
         f"🔍 View metrics for your job via "
         f"[cyan]'truss train metrics --job-id {job_id}'[/cyan]\n"
+        f"{cache_summary_snippet}"
         f"🌐 Status page: {common.format_link(core.status_page_url(remote_provider.remote_url, job_id))}"
     )
 
@@ -44,6 +69,7 @@ def _handle_post_create_logic(
     job_resp: dict, remote_provider: BasetenRemote, tail: bool
 ) -> None:
     project_id, job_id = job_resp["training_project"]["id"], job_resp["id"]
+    project_name = job_resp["training_project"]["name"]
 
     if job_resp.get("current_status", None) == "TRAINING_JOB_QUEUED":
         console.print(
@@ -51,7 +77,9 @@ def _handle_post_create_logic(
             style="green",
         )
     else:
-        _print_training_job_success_message(job_id, remote_provider)
+        _print_training_job_success_message(
+            job_id, project_name, job_resp["job_object"], remote_provider
+        )
 
     if tail:
         watcher = TrainingLogWatcher(remote_provider.api, project_id, job_id)
@@ -351,3 +379,46 @@ def download_checkpoint_artifacts(job_id: Optional[str], remote: Optional[str])
     except Exception as e:
         error_console.print(f"Failed to download checkpoint artifacts data: {str(e)}")
         sys.exit(1)
+
+
+@train.group(name="cache")
+def cache():
+    """Cache-related subcommands for truss train"""
+
+
+@cache.command(name="summarize")
+@click.option(
+    "--project", type=str, required=True, help="Project ID or name to view cache for."
+)
+@click.option("--remote", type=str, required=False, help="Remote to use")
+@click.option(
+    "--sort",
+    type=click.Choice(
+        [
+            SORT_BY_FILEPATH,
+            SORT_BY_SIZE,
+            SORT_BY_MODIFIED,
+            SORT_BY_TYPE,
+            SORT_BY_PERMISSIONS,
+        ]
+    ),
+    default=SORT_BY_FILEPATH,
+    help="Sort files by filepath, size, modified date, file type, or permissions.",
+)
+@click.option(
+    "--order",
+    type=click.Choice([SORT_ORDER_ASC, SORT_ORDER_DESC]),
+    default=SORT_ORDER_ASC,
+    help="Sort order: ascending or descending.",
+)
+@common.common_options()
+def view_cache_summary(project: str, remote: Optional[str], sort: str, order: str):
+    """View cache summary for a training project"""
+    if not remote:
+        remote = remote_cli.inquire_remote_name()
+
+    remote_provider: BasetenRemote = cast(
+        BasetenRemote, RemoteFactory.create(remote=remote)
+    )
+
+    train_cli.view_cache_summary_by_project(remote_provider, project, sort, order)

truss/cli/utils/common.py

@@ -186,4 +186,17 @@ def format_localized_time(iso_timestamp: str) -> str:
         iso_timestamp = iso_timestamp.replace("Z", "+00:00")
     utc_time = datetime.datetime.fromisoformat(iso_timestamp)
     local_time = utc_time.astimezone()
-    return local_time.strftime("%Y-%m-%d %H:%M")
+    return local_time.strftime("%Y-%m-%d %H:%M:%S")
+
+
+def format_bytes_to_human_readable(bytes: int) -> str:
+    if bytes > 1000 * 1000 * 1000 * 1000:
+        return f"{bytes / (1000 * 1000 * 1000 * 1000):.2f} TB"
+    if bytes > 1000 * 1000 * 1000:
+        return f"{bytes / (1000 * 1000 * 1000):.2f} GB"
+    elif bytes > 1000 * 1000:
+        return f"{bytes / (1000 * 1000):.2f} MB"
+    elif bytes > 1000:
+        return f"{bytes / 1000:.2f} KB"
+    else:
+        return f"{bytes} B"

truss/remote/baseten/api.py

@@ -669,6 +669,13 @@ class BasetenApi:
         # NB(nikhil): reverse order so latest logs are at the end
         return resp_json["logs"][::-1]
 
+    def get_cache_summary(self, project_id: str):
+        """Get cache summary for a training project."""
+        resp_json = self._rest_api_client.get(
+            f"v1/training_projects/{project_id}/cache/summary"
+        )
+        return resp_json
+
     def _fetch_log_batch(
         self, project_id: str, job_id: str, query_params: Dict[str, Any]
     ) -> List[Any]:

truss/remote/baseten/custom_types.py

@@ -120,3 +120,31 @@ class TrussUserEnv(pydantic.BaseModel):
 class BlobType(Enum):
     MODEL = "model"
     TRAIN = "train"
+
+
+class FileSummary(pydantic.BaseModel):
+    """Information about a file in the cache."""
+
+    path: str = pydantic.Field(description="Relative path of the file in the cache")
+    size_bytes: int = pydantic.Field(description="Size of the file in bytes")
+    modified: str = pydantic.Field(description="Last modification time of the file")
+    file_type: Optional[str] = pydantic.Field(
+        default=None,
+        description="Type of the file (e.g., 'file', 'directory', 'symlink')",
+    )
+    permissions: Optional[str] = pydantic.Field(
+        default=None,
+        description="File permissions in Unix symbolic format (e.g., 'drwxr-xr-x', '-rw-r--r--')",
+    )
+
+
+class GetCacheSummaryResponseV1(pydantic.BaseModel):
+    """Response for getting cache summary."""
+
+    timestamp: str = pydantic.Field(
+        description="Timestamp when the cache summary was captured"
+    )
+    project_id: str = pydantic.Field(description="Project ID associated with the cache")
+    file_summaries: list[FileSummary] = pydantic.Field(
+        description="List of files in the cache"
+    )

truss/tests/cli/train/test_cache_view.py

@@ -0,0 +1,712 @@
+from unittest.mock import Mock
+
+import click
+import pytest
+
+from truss.cli.train.core import (
+    SORT_BY_FILEPATH,
+    SORT_BY_MODIFIED,
+    SORT_BY_PERMISSIONS,
+    SORT_BY_SIZE,
+    SORT_BY_TYPE,
+    SORT_ORDER_ASC,
+    SORT_ORDER_DESC,
+    view_cache_summary,
+    view_cache_summary_by_project,
+)
+from truss.remote.baseten.remote import BasetenRemote
+
+
+def test_view_cache_summary_success(capsys):
+    """Test successful cache summary viewing."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "model/weights.bin",
+                "size_bytes": 1024 * 1024 * 100,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "config.json",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "model/",
+                "size_bytes": 0,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "directory",
+                "permissions": "drwxr-xr-x",
+            },
+        ],
+    }
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_FILEPATH, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    assert "Cache summary for project: proj123" in captured.out
+    assert "model/weights.bin" in captured.out
+    assert "config.json" in captured.out
+    assert "104.86 MB" in captured.out
+    assert "1.02 KB" in captured.out
+
+
+def test_view_cache_summary_no_cache(capsys):
+    """Test when no cache summary is found."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.get_cache_summary.return_value = {}
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_FILEPATH, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    assert "No cache summary found for this project." in captured.out
+
+
+def test_view_cache_summary_empty_files(capsys):
+    """Test when cache summary exists but has no files."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_FILEPATH, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    assert "No files found in cache." in captured.out
+
+
+def test_view_cache_summary_api_error(capsys):
+    """Test when API call fails."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.side_effect = Exception("API Error")
+
+    with pytest.raises(Exception, match="API Error"):
+        view_cache_summary(mock_remote, "proj123", SORT_BY_FILEPATH, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    assert "Error fetching cache summary: API Error" in captured.out
+
+
+def test_view_cache_summary_sort_by_size_asc(capsys):
+    """Test sorting by size in ascending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "large_file.bin",
+                "size_bytes": 1024 * 1024 * 100,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "small_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "medium_file.dat",
+                "size_bytes": 1024 * 1024,
+                "modified": "2024-01-01T11:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_SIZE, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    output_lines = captured.out.split("\n")
+
+    table_start = -1
+    for i, line in enumerate(output_lines):
+        if "small_file.txt" in line:
+            table_start = i
+            break
+
+    small_file_line = None
+    medium_file_line = None
+    large_file_line = None
+
+    for line in output_lines[table_start:]:
+        if "small_file.txt" in line:
+            small_file_line = line
+        elif "medium_file.dat" in line:
+            medium_file_line = line
+        elif "large_file.bin" in line:
+            large_file_line = line
+
+    assert small_file_line is not None
+    assert medium_file_line is not None
+    assert large_file_line is not None
+
+    small_pos = captured.out.find("small_file.txt")
+    medium_pos = captured.out.find("medium_file.dat")
+    large_pos = captured.out.find("large_file.bin")
+
+    assert small_pos < medium_pos < large_pos
+
+
+def test_view_cache_summary_sort_by_size_desc(capsys):
+    """Test sorting by size in descending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "large_file.bin",
+                "size_bytes": 1024 * 1024 * 100,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "small_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "medium_file.dat",
+                "size_bytes": 1024 * 1024,
+                "modified": "2024-01-01T11:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_SIZE, SORT_ORDER_DESC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    small_pos = captured.out.find("small_file.txt")
+    medium_pos = captured.out.find("medium_file.dat")
+    large_pos = captured.out.find("large_file.bin")
+
+    assert large_pos < medium_pos < small_pos
+
+
+def test_view_cache_summary_sort_by_modified_asc(capsys):
+    """Test sorting by modified date in ascending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "old_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "new_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T12:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "middle_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_MODIFIED, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    old_pos = captured.out.find("old_file.txt")
+    middle_pos = captured.out.find("middle_file.txt")
+    new_pos = captured.out.find("new_file.txt")
+
+    assert old_pos < middle_pos < new_pos
+
+
+def test_view_cache_summary_sort_by_filepath_desc(capsys):
+    """Test sorting by filepath in descending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "a_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "z_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "m_file.txt",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_FILEPATH, SORT_ORDER_DESC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    a_pos = captured.out.find("a_file.txt")
+    m_pos = captured.out.find("m_file.txt")
+    z_pos = captured.out.find("z_file.txt")
+
+    assert z_pos < m_pos < a_pos
+
+
+def test_view_cache_summary_by_project_name_success(capsys):
+    """Test successful cache summary viewing by project name."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    # Mock the get_cache_summary response for successful project ID lookup
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "model/weights.bin",
+                "size_bytes": 1024 * 1024 * 100,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            }
+        ],
+    }
+
+    # Mock the list_training_projects response
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"},
+        {"id": "proj456", "name": "another-project"},
+    ]
+
+    view_cache_summary_by_project(
+        mock_remote, "test-project", SORT_BY_FILEPATH, SORT_ORDER_ASC
+    )
+
+    assert mock_api.get_cache_summary.call_count == 1
+    assert mock_api.list_training_projects.call_count == 1
+
+    captured = capsys.readouterr()
+    assert "Cache summary for project: proj123" in captured.out
+    assert "model/weights.bin" in captured.out
+
+
+def test_view_cache_summary_by_project_name_not_found(capsys):
+    """Test when project name is not found."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"},
+        {"id": "proj456", "name": "another-project"},
+    ]
+
+    with pytest.raises(
+        click.ClickException, match="Project 'nonexistent-project' not found"
+    ):
+        view_cache_summary_by_project(
+            mock_remote, "nonexistent-project", SORT_BY_FILEPATH, SORT_ORDER_ASC
+        )
+
+    mock_api.list_training_projects.assert_called_once()
+
+
+def test_view_cache_summary_by_project_id_direct(capsys):
+    """Test that project ID is used directly."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "model/weights.bin",
+                "size_bytes": 1024 * 1024 * 100,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            }
+        ],
+    }
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"},
+        {"id": "proj456", "name": "another-project"},
+    ]
+
+    project_id = "proj123"
+    view_cache_summary_by_project(
+        mock_remote, project_id, SORT_BY_FILEPATH, SORT_ORDER_ASC
+    )
+
+    assert mock_api.get_cache_summary.call_count == 1
+    assert mock_api.list_training_projects.call_count == 1
+
+    captured = capsys.readouterr()
+    assert "Cache summary for project: proj123" in captured.out
+
+
+def test_view_cache_summary_by_project_other_error():
+    """Test that other errors (not 404) are re-raised."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "some-project"}
+    ]
+    mock_api.get_cache_summary.side_effect = Exception("Network error")
+
+    with pytest.raises(Exception, match="Network error"):
+        view_cache_summary_by_project(
+            mock_remote, "some-project", SORT_BY_FILEPATH, SORT_ORDER_ASC
+        )
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+
+def test_view_cache_summary_by_project_list_error(capsys):
+    """Test when listing projects fails after 404."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.side_effect = Exception("API error")
+
+    with pytest.raises(click.ClickException, match="Error fetching project: API error"):
+        view_cache_summary_by_project(
+            mock_remote, "nonexistent-project", SORT_BY_FILEPATH, SORT_ORDER_ASC
+        )
+
+    mock_api.list_training_projects.assert_called_once()
+
+
+def test_view_cache_summary_sort_by_type_asc(capsys):
+    """Test sorting by file type in ascending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "config.json",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "model/",
+                "size_bytes": 0,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "directory",
+                "permissions": "drwxr-xr-x",
+            },
+            {
+                "path": "data.txt",
+                "size_bytes": 2048,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_TYPE, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+    output_lines = captured.out.split("\n")
+
+    table_start = -1
+    for i, line in enumerate(output_lines):
+        if "config.json" in line or "data.txt" in line or "model/" in line:
+            table_start = i
+            break
+
+    directory_line = None
+    file_lines = []
+
+    for line in output_lines[table_start:]:
+        if "model/" in line:
+            directory_line = line
+        elif "config.json" in line or "data.txt" in line:
+            file_lines.append(line)
+
+    assert directory_line is not None
+    assert len(file_lines) == 2
+
+    directory_pos = captured.out.find("model/")
+    config_pos = captured.out.find("config.json")
+    data_pos = captured.out.find("data.txt")
+
+    assert directory_pos < config_pos
+    assert directory_pos < data_pos
+
+
+def test_view_cache_summary_sort_by_type_desc(capsys):
+    """Test sorting by file type in descending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "config.json",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "model/",
+                "size_bytes": 0,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "directory",
+                "permissions": "drwxr-xr-x",
+            },
+            {
+                "path": "data.txt",
+                "size_bytes": 2048,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_TYPE, SORT_ORDER_DESC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    config_pos = captured.out.find("config.json")
+    data_pos = captured.out.find("data.txt")
+    directory_pos = captured.out.find("model/")
+
+    assert config_pos < directory_pos
+    assert data_pos < directory_pos
+
+
+def test_view_cache_summary_sort_by_permissions_asc(capsys):
+    """Test sorting by permissions in ascending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "config.json",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "model/",
+                "size_bytes": 0,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "directory",
+                "permissions": "drwxr-xr-x",
+            },
+            {
+                "path": "script.sh",
+                "size_bytes": 2048,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rwxr-xr-x",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_PERMISSIONS, SORT_ORDER_ASC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    config_pos = captured.out.find("config.json")
+    script_pos = captured.out.find("script.sh")
+    directory_pos = captured.out.find("model/")
+
+    assert config_pos != -1, "Config file not found in output"
+    assert script_pos != -1, "Script file not found in output"
+    assert directory_pos != -1, "Directory not found in output"
+
+    assert config_pos < script_pos
+    assert script_pos < directory_pos
+
+
+def test_view_cache_summary_sort_by_permissions_desc(capsys):
+    """Test sorting by permissions in descending order."""
+    mock_api = Mock()
+    mock_remote = Mock(spec=BasetenRemote)
+    mock_remote.api = mock_api
+
+    mock_api.list_training_projects.return_value = [
+        {"id": "proj123", "name": "test-project"}
+    ]
+    mock_api.get_cache_summary.return_value = {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "project_id": "proj123",
+        "file_summaries": [
+            {
+                "path": "config.json",
+                "size_bytes": 1024,
+                "modified": "2024-01-01T09:00:00Z",
+                "file_type": "file",
+                "permissions": "-rw-r--r--",
+            },
+            {
+                "path": "model/",
+                "size_bytes": 0,
+                "modified": "2024-01-01T08:00:00Z",
+                "file_type": "directory",
+                "permissions": "drwxr-xr-x",
+            },
+            {
+                "path": "script.sh",
+                "size_bytes": 2048,
+                "modified": "2024-01-01T10:00:00Z",
+                "file_type": "file",
+                "permissions": "-rwxr-xr-x",
+            },
+        ],
+    }
+
+    view_cache_summary(mock_remote, "proj123", SORT_BY_PERMISSIONS, SORT_ORDER_DESC)
+
+    mock_api.get_cache_summary.assert_called_once_with("proj123")
+
+    captured = capsys.readouterr()
+
+    directory_pos = captured.out.find("model/")
+    script_pos = captured.out.find("script.sh")
+    config_pos = captured.out.find("config.json")
+
+    assert directory_pos != -1, "Directory not found in output"
+    assert script_pos != -1, "Script file not found in output"
+    assert config_pos != -1, "Config file not found in output"
+
+    assert directory_pos < script_pos
+    assert script_pos < config_pos

{truss-0.10.11rc2.dist-info → truss-0.10.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.10.11rc2
+Version: 0.10.12
 Summary: A seamless bridge from model development to model delivery
 Project-URL: Repository, https://github.com/basetenlabs/truss
 Project-URL: Homepage, https://truss.baseten.co

{truss-0.10.11rc2.dist-info → truss-0.10.12.dist-info}/RECORD

@@ -11,16 +11,16 @@ truss/base/truss_spec.py,sha256=jFVF79CXoEEspl2kXBAPyi-rwISReIGTdobGpaIhwJw,5979
 truss/cli/chains_commands.py,sha256=y6pdIAGCcKOPG9bPuCXPfSA0onQm5x-tT_3blSBfPYg,16971
 truss/cli/cli.py,sha256=PaMkuwXZflkU7sa1tEoT_Zmy-iBkEZs1m4IVqcieaeo,30367
 truss/cli/remote_cli.py,sha256=G_xCKRXzgkCmkiZJhUFfsv5YSVgde1jLA5LPQitpZgI,1905
-truss/cli/train_commands.py,sha256=P9bdnpq1SgEGXBaVf9joKdsaCDX2v29P4MhLMuz-jYw,12344
+truss/cli/train_commands.py,sha256=eFVx8Lba4ILYA4FCjRjMmABH-uUR_i7h4Io1mn7reZM,14423
 truss/cli/logs/base_watcher.py,sha256=KKyd7lIrdaEeDVt8EtjMioSPGVpLyOcF0ewyzE_GGdQ,2785
 truss/cli/logs/model_log_watcher.py,sha256=NACcP-wkcaroYa2Cb9BZC7Yr0554WZa_FSM2LXOf4A8,1263
 truss/cli/logs/training_log_watcher.py,sha256=r6HRqrLnz-PiKTUXiDYYxg4ZnP8vYcXlEX1YmgHhzlo,1173
 truss/cli/logs/utils.py,sha256=z-U_FG4BUzdZLbE3BnXb4DZQ0zt3LSZ3PiQpLaDuc3o,1031
-truss/cli/train/common.py,sha256=Es1yllSYxjM9x2uBzTGbYwyd8ML66cqqge0XO8_G_X0,992
-truss/cli/train/core.py,sha256=MBOhPSVYOU7wVh09uWQrJDEVOhJQug_2Odv3u6tCVTA,13855
+truss/cli/train/common.py,sha256=xTR41U5FeSndXfNBBHF9wF5XwZH1sOIVFlv-XHjsKIU,1547
+truss/cli/train/core.py,sha256=gfFRqxCxGsanbMNgJBJ0WFqVxZm4SwOEM2iyLS3K8Ns,18687
 truss/cli/train/deploy_from_checkpoint_config.yml,sha256=mktaVrfhN8Kjx1UveC4xr-gTW-kjwbHvq6bx_LpO-Wg,371
 truss/cli/train/deploy_from_checkpoint_config_whisper.yml,sha256=6GbOorYC8ml0UyOUvuBpFO_fuYtYE646JqsalR-D4oY,406
-truss/cli/train/metrics_watcher.py,sha256=ftrLQ5m7V1lAqcAvdGbMv5r0aF4D0lypfKjokCBQvLw,12798
+truss/cli/train/metrics_watcher.py,sha256=smz-zrEsBj_-wJHI0pAZ-EAPrvfCWzq1eQjGiFNM-Mk,12755
 truss/cli/train/poller.py,sha256=TGRzELxsicga0bEXewSX1ujw6lfPmDnHd6nr8zvOFO8,3550
 truss/cli/train/types.py,sha256=alGtr4Q71GeB65PpGMhsoKygw4k_ncR6MKIP1ioP8rI,951
 truss/cli/train/deploy_checkpoints/__init__.py,sha256=wL-M2yu8PxO2tFvjwshXAfPnB-5TlvsBp2v_bdzimRU,99
@@ -29,7 +29,7 @@ truss/cli/train/deploy_checkpoints/deploy_checkpoints_helpers.py,sha256=6x5nS_Hn
 truss/cli/train/deploy_checkpoints/deploy_full_checkpoints.py,sha256=FYRG5KTMlxEMZS-RA_m2gp1wuqWbSpqt2RhdQfLibhA,3968
 truss/cli/train/deploy_checkpoints/deploy_lora_checkpoints.py,sha256=P91dIAzuhl2GlzmrWwCcYI7uCMT1Lm7C79JQHM_exN4,4442
 truss/cli/train/deploy_checkpoints/deploy_whisper_checkpoints.py,sha256=NSo2kEn-CxawGvUhn-xE81noxoTJ0cCv8X9fkrMxAsM,2617
-truss/cli/utils/common.py,sha256=aWnla4qMSEz57dRMTl7R-EaScsuEpnQUeziGUaIeqeU,6149
+truss/cli/utils/common.py,sha256=ink9ZE0MsOv6PCFK_Ra5k1aHm281TXTnMpnLjf2PtUM,6585
 truss/cli/utils/output.py,sha256=GNjU85ZAMp5BI6Yij5wYXcaAvpm_kmHV0nHNmdkMxb0,646
 truss/cli/utils/self_upgrade.py,sha256=eTJZA4Wc8uUp4Qh6viRQp6bZm--wnQp7KWe5KRRpPtg,5427
 truss/contexts/docker_build_setup.py,sha256=cF4ExZgtYvrWxvyCAaUZUvV_DB_7__MqVomUDpalvKo,3925
@@ -52,10 +52,10 @@ truss/patch/truss_dir_patch_applier.py,sha256=ALnaVnu96g0kF2UmGuBFTua3lrXpwAy4sG
 truss/remote/remote_factory.py,sha256=-0gLh_yIyNDgD48Q6sR8Yo5dOMQg84lrHRvn_XR0n4s,3585
 truss/remote/truss_remote.py,sha256=TEe6h6by5-JLy7PMFsDN2QxIY5FmdIYN3bKvHHl02xM,8440
 truss/remote/baseten/__init__.py,sha256=XNqJW1zyp143XQc6-7XVwsUA_Q_ZJv_ausn1_Ohtw9Y,176
-truss/remote/baseten/api.py,sha256=6Nie4hv4z5I62boeCQvP3tGA0Pwu96bMgz1vp5Tkxao,24447
+truss/remote/baseten/api.py,sha256=lJOt2i3tu0ZeCh4B_-hpfpjcZKgTHVnkxraooK7TUHw,24699
 truss/remote/baseten/auth.py,sha256=tI7s6cI2EZgzpMIzrdbILHyGwiHDnmoKf_JBhJXT55E,776
 truss/remote/baseten/core.py,sha256=uxtmBI9RAVHu1glIEJb5Q4ccJYLeZM1Cp5Svb9W68Yw,21965
-truss/remote/baseten/custom_types.py,sha256=g7MwgYaeqIxF-e170G5iEVLWiw5jgAnqXztIUqVkdyc,3227
+truss/remote/baseten/custom_types.py,sha256=gUG7EkTeXzqcqznbKxz1SGTtHavNKGm1UoTFiln3LmQ,4309
 truss/remote/baseten/error.py,sha256=3TNTwwPqZnr4NRd9Sl6SfLUQR2fz9l6akDPpOntTpzA,578
 truss/remote/baseten/remote.py,sha256=Se8AES5mk8jxa8S9fN2DSG7wnsaV7ftRjJ4Uwc_w_S0,22544
 truss/remote/baseten/rest_client.py,sha256=_t3CWsWARt2u0C0fDsF4rtvkkHe-lH7KXoPxWXAkKd4,1185
@@ -139,6 +139,7 @@ truss/tests/test_truss_gatherer.py,sha256=bn288OEkC49YY0mhly4cAl410ktZPfElNdWwZy
 truss/tests/test_truss_handle.py,sha256=-xz9VXkecXDTslmQZ-dmUmQLnvD0uumRqHS2uvGlMBA,30750
 truss/tests/test_util.py,sha256=hs1bNMkXKEdoPRx4Nw-NAEdoibR92OubZuADGmbiYsQ,1344
 truss/tests/cli/test_cli.py,sha256=yfbVS5u1hnAmmA8mJ539vj3lhH-JVGUvC4Q_Mbort44,787
+truss/tests/cli/train/test_cache_view.py,sha256=aVRCh3atRpFbJqyYgq7N-vAW0DiKMftQ7ajUqO2ClOg,22606
 truss/tests/cli/train/test_deploy_checkpoints.py,sha256=wQZ3DPLPAyXE3iaQiyHJTBO15v_gXN44eDk1StYkKmM,44764
 truss/tests/cli/train/test_train_cli_core.py,sha256=T1Xa6-NRk2nTJGX6sXaA8x4qCwL3Ini72PBI2gW7rYM,7879
 truss/tests/cli/train/resources/test_deploy_from_checkpoint_config.yml,sha256=GF7r9l0KaeXiUYCPSBpeMPd2QG6PeWWyI12NdbqLOgc,1930
@@ -360,11 +361,11 @@ truss_chains/remote_chainlet/stub.py,sha256=Y2gDUzMY9WRaQNHIz-o4dfLUfFyYV9dUhIRQ
 truss_chains/remote_chainlet/utils.py,sha256=O_5P-VAUvg0cegEW1uKCOf5EBwD8rEGYVoGMivOmc7k,22374
 truss_train/__init__.py,sha256=7hE6j6-u6UGzCGaNp3CsCN0kAVjBus1Ekups-Bk0fi4,837
 truss_train/definitions.py,sha256=V985HhY4rdXL10DZxpFEpze9ScxzWErMht4WwaPknGU,6789
-truss_train/deployment.py,sha256=fDYRfzFRtVKMRVG0bKXYPmx6HXwLE0ukSQ0f81hG8kk,3020
+truss_train/deployment.py,sha256=lWWANSuzBWu2M4oK4qD7n-oVR1JKdmw2Pn5BJQHg-Ck,3074
 truss_train/loader.py,sha256=0o66EjBaHc2YY4syxxHVR4ordJWs13lNXnKjKq2wq0U,1630
 truss_train/public_api.py,sha256=9N_NstiUlmBuLUwH_fNG_1x7OhGCytZLNvqKXBlStrM,1220
-truss-0.10.11rc2.dist-info/METADATA,sha256=rEMcJZISmCLDeN8M9pcdUFXyiTxeERavaJszW3ZlDz0,6673
-truss-0.10.11rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truss-0.10.11rc2.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
-truss-0.10.11rc2.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
-truss-0.10.11rc2.dist-info/RECORD,,
+truss-0.10.12.dist-info/METADATA,sha256=D39-4SrfGGxR5CjGVA9z4w8f9yB4JODhNM1sowDnnmw,6670
+truss-0.10.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truss-0.10.12.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
+truss-0.10.12.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
+truss-0.10.12.dist-info/RECORD,,

truss_train/deployment.py

@@ -83,4 +83,5 @@ def create_training_job_from_file(
             training_project=training_project,
             config=config,
         )
+        job_resp["job_object"] = training_project.job
     return job_resp