outerbounds 0.3.182rc0__py3-none-any.whl → 0.3.182rc2__py3-none-any.whl

outerbounds/apps/_state_machine.py

@@ -342,17 +342,22 @@ def _capsule_worker_semantic_status(
 
     status_dict: CapsuleWorkerStatusDict = {
         "at_least_one_pending": count_for_version(pending_workers) > 0,
-        "at_least_one_running": count_for_version(running_workers) > 0,
+        # if min_replicas is 0, the at_least_one_running should be true for running worker count = 0
+        "at_least_one_running": (
+            count_for_version(running_workers) >= min(min_replicas, 1)
+        ),
         "at_least_one_crashlooping": count_for_version(crashlooping_workers) > 0,
         "none_present": (
             count_for_version(running_workers) == 0
             and count_for_version(pending_workers) == 0
             and count_for_version(crashlooping_workers) == 0
         ),
-        "all_running": count_for_version(running_workers) == min_replicas,
-        "fully_finished": count_for_version(running_workers) == min_replicas
-        and len(pending_workers) == 0
-        and len(crashlooping_workers) == 0,
+        "all_running": count_for_version(running_workers) >= min_replicas,
+        "fully_finished": (
+            count_for_version(running_workers) >= min_replicas
+            and len(pending_workers) == 0
+            and len(crashlooping_workers) == 0
+        ),
         "current_info": {
             "pending": count_for_version(pending_workers),
             "running": count_for_version(running_workers),

outerbounds/apps/app_cli.py

@@ -439,6 +439,12 @@ def deployment_instance_options(func):
         help=DEPLOYMENT_READY_CONDITIONS.__doc__,
         default=DEPLOYMENT_READY_CONDITIONS.ATLEAST_ONE_RUNNING,
     )
+    @click.option(
+        "--status-file",
+        type=str,
+        help="The path to the file where the final status of the deployment will be written.",
+        default=None,
+    )
     @click.option(
         "--readiness-wait-time",
         type=int,
@@ -446,9 +452,10 @@ def deployment_instance_options(func):
         default=4,
     )
     @click.option(
-        "--max-wait-time",
+        "--deployment-timeout",
+        "max_wait_time",
         type=int,
-        help="The maximum time (in seconds) to wait for the deployment to be ready.",
+        help="The maximum time (in seconds) to wait for the deployment to reach readiness before timing out.",
         default=600,
     )
     @click.option(
@@ -702,6 +709,7 @@ def deploy(
     readiness_condition=None,
     max_wait_time=None,
     readiness_wait_time=None,
+    status_file=None,
     no_loader=False,
     **options,
 ):
@@ -908,7 +916,9 @@ def deploy(
             capsule_logger = partial(_spinner_logger, capsule_spinner)
             capsule_spinner.start()
 
-        capsule.wait_for_terminal_state(logger=capsule_logger)
+        # We only get the `capsule_response` if the deployment is has reached
+        # a successful terminal state.
+        final_status = capsule.wait_for_terminal_state(logger=capsule_logger)
         if capsule_spinner:
             capsule_spinner.stop()
 
@@ -918,6 +928,16 @@ def deploy(
                 system_msg=True,
             )
 
+        if status_file:
+            # Create the file if it doesn't exist
+            with open(status_file, "w") as f:
+                f.write(json.dumps(final_status, indent=4))
+            logger(
+                f"📝 {capsule.capsule_type} {app_config.config['name']} ({capsule.identifier}) deployment status written to {status_file}",
+                color=ColorTheme.INFO_COLOR,
+                system_msg=True,
+            )
+
     except Exception as e:
         logger(
             f"Deployment failed: [{e.__class__.__name__}]: {e}",

outerbounds/apps/capsule.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 import json
 import os
 import pathlib
@@ -751,7 +752,8 @@ class CapsuleDeployer:
                     # side will be really helpful.
                     self._monitor_worker_readiness(workers_state_machine, logger)
 
-                elif workers_state_machine.is_failure_state:
+                # We should still check for failure state and crash if we detect something in the readiness check
+                if workers_state_machine.is_failure_state:
                     # hit the logs endpoint for the worker and get the logs
                     # Print those logs out on the terminal
                     # raise an exception that should be caught gracefully by the cli
@@ -795,4 +797,13 @@ class CapsuleDeployer:
                 self.identifier,
                 f"Capsule {self.identifier} failed to be ready to serve traffic",
             )
-        return capsule_response
+
+        return dict(
+            id=self.identifier,
+            auth_type=self.capsule_type,
+            public_url=self.status.out_of_cluster_url,
+            available_replicas=self.status.available_replicas,
+            name=self.name,
+            deployed_version=self.current_deployment_instance_version,
+            deployed_at=datetime.now().isoformat(),
+        )

{outerbounds-0.3.182rc0.dist-info → outerbounds-0.3.182rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: outerbounds
-Version: 0.3.182rc0
+Version: 0.3.182rc2
 Summary: More Data Science, Less Administration
 License: Proprietary
 Keywords: data science,machine learning,MLOps
@@ -29,8 +29,8 @@ Requires-Dist: google-cloud-secret-manager (>=2.20.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: metaflow-checkpoint (==0.2.1)
 Requires-Dist: ob-metaflow (==2.15.17.1)
-Requires-Dist: ob-metaflow-extensions (==1.1.170rc0)
-Requires-Dist: ob-metaflow-stubs (==6.0.3.182rc0)
+Requires-Dist: ob-metaflow-extensions (==1.1.170rc2)
+Requires-Dist: ob-metaflow-stubs (==6.0.3.182rc2)
 Requires-Dist: opentelemetry-distro (>=0.41b0) ; extra == "otel"
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.20.0) ; extra == "otel"
 Requires-Dist: opentelemetry-instrumentation-requests (>=0.41b0) ; extra == "otel"

{outerbounds-0.3.182rc0.dist-info → outerbounds-0.3.182rc2.dist-info}/RECORD

@@ -43,11 +43,11 @@ outerbounds/_vendor/yaml/scanner.py,sha256=ZcI8IngR56PaQ0m27WU2vxCqmDCuRjz-hr7pi
 outerbounds/_vendor/yaml/serializer.py,sha256=8wFZRy9SsQSktF_f9OOroroqsh4qVUe53ry07P9UgCc,4368
 outerbounds/_vendor/yaml/tokens.py,sha256=JBSu38wihGr4l73JwbfMA7Ks1-X84g8-NskTz7KwPmA,2578
 outerbounds/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-outerbounds/apps/_state_machine.py,sha256=ixgL--jne3q71gQNnUeK-UdLP-Oc2kSGSsIW1fiHZPY,14469
-outerbounds/apps/app_cli.py,sha256=vTbIN43A9A_OHTM0I6cb28g3GtY4_jk1wmUi5wG09w0,50174
+outerbounds/apps/_state_machine.py,sha256=eAJFNt_9kwE70XyXoMU107TrzPRnb2xcAyVtXV1yAGc,14646
+outerbounds/apps/app_cli.py,sha256=0TCSxkKz9psJMGULMJcqXvrEJNibBDUTVoY37N5pcIk,51003
 outerbounds/apps/app_config.py,sha256=UHVK8JLIuW-OcGg5WxDm4QHeImPGtohD4KpJryZntC4,11307
 outerbounds/apps/artifacts.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-outerbounds/apps/capsule.py,sha256=IYaD5X-IpSD9IsLpIJUaJ2B31up7PY7GhMMizkwLV7I,31039
+outerbounds/apps/capsule.py,sha256=8ZoEjLwcUvbps6H03hkl25cAGj8pyaVNziaAK8AZTxY,31523
 outerbounds/apps/cli_to_config.py,sha256=Thc5jXRxoU6Pr8kAVVOX-5Es5ha6y6Vh_GBzL__oI7Q,3299
 outerbounds/apps/click_importer.py,sha256=nnkPOR6TKrtIpc3a5Fna1zVJoQqDZvUXlNA9CdiNKFc,995
 outerbounds/apps/code_package/__init__.py,sha256=8McF7pgx8ghvjRnazp2Qktlxi9yYwNiwESSQrk-2oW8,68
@@ -78,7 +78,7 @@ outerbounds/utils/metaflowconfig.py,sha256=l2vJbgPkLISU-XPGZFaC8ZKmYFyJemlD6bwB-
 outerbounds/utils/schema.py,sha256=lMUr9kNgn9wy-sO_t_Tlxmbt63yLeN4b0xQXbDUDj4A,2331
 outerbounds/utils/utils.py,sha256=4Z8cszNob_8kDYCLNTrP-wWads_S_MdL3Uj3ju4mEsk,501
 outerbounds/vendor.py,sha256=gRLRJNXtZBeUpPEog0LOeIsl6GosaFFbCxUvR4bW6IQ,5093
-outerbounds-0.3.182rc0.dist-info/METADATA,sha256=Y51QdSpzQvPHckh4lLqNcic0u-gM1sIi4rX2K4RmUPA,1846
-outerbounds-0.3.182rc0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-outerbounds-0.3.182rc0.dist-info/entry_points.txt,sha256=AP6rZg7y5SK9e9a9iVq0Fi9Q2KPjPZSwtZ6R98rLw-8,56
-outerbounds-0.3.182rc0.dist-info/RECORD,,
+outerbounds-0.3.182rc2.dist-info/METADATA,sha256=QNhNeHMNuCMReA27whbsen3OPIniLUinbQ3ScvLfQLc,1846
+outerbounds-0.3.182rc2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+outerbounds-0.3.182rc2.dist-info/entry_points.txt,sha256=AP6rZg7y5SK9e9a9iVq0Fi9Q2KPjPZSwtZ6R98rLw-8,56
+outerbounds-0.3.182rc2.dist-info/RECORD,,