outerbounds 0.9.0__py3-none-any.whl → 0.9.1__py3-none-any.whl

outerbounds/command_groups/perimeters_cli.py

@@ -126,7 +126,7 @@ def switch(config_dir=None, profile=None, output="", id=None, force=False):
     )
 
     try:
-        ensure_cloud_credentials_for_shell(config_dir, profile)
+        ensure_cloud_credentials_for_shell(config_dir, profile, "")
     except:
         click.secho(
             "Failed to update cloud credentials.",
@@ -314,7 +314,13 @@ def list(config_dir=None, profile=None, output=""):
     help="Show output in the specified format.",
     type=click.Choice(["json", ""]),
 )
-def ensure_cloud_creds(config_dir=None, profile=None, output=""):
+@click.option(
+    "-c",
+    "--cspr-override",
+    default="",
+    help="Override the CSPR role ARN to use for AWS credentials.",
+)
+def ensure_cloud_creds(config_dir=None, profile=None, output="", cspr_override=""):
     ensure_cloud_creds_step = CommandStatus(
         "EnsureCloudCredentials",
         OuterboundsCommandStatus.OK,
@@ -324,7 +330,7 @@ def ensure_cloud_creds(config_dir=None, profile=None, output=""):
     ensure_cloud_creds_response = OuterboundsCommandResponse()
 
     try:
-        ensure_cloud_credentials_for_shell(config_dir, profile)
+        ensure_cloud_credentials_for_shell(config_dir, profile, cspr_override)
         click.secho("Cloud credentials updated successfully.", fg="green", err=True)
     except:
         click.secho(
@@ -430,7 +436,7 @@ def get_ob_config_or_fail_command(
     return ob_config_dict
 
 
-def ensure_cloud_credentials_for_shell(config_dir, profile):
+def ensure_cloud_credentials_for_shell(config_dir, profile, cspr_override):
     if "WORKSTATION_ID" not in os.environ:
         # Naive check to see if we're running in workstation. No need to ensure anything
         # if this is not a workstation.
@@ -444,7 +450,7 @@ def ensure_cloud_credentials_for_shell(config_dir, profile):
         ensure_gcp_cloud_creds(config_dir, profile)
     elif "METAFLOW_DEFAULT_AWS_CLIENT_PROVIDER" in mf_config:
         # This is an AWS deployment.
-        ensure_aws_cloud_creds(config_dir, profile)
+        ensure_aws_cloud_creds(config_dir, profile, cspr_override)
 
 
 def confirm_user_has_access_to_perimeter_or_fail(
@@ -502,7 +508,7 @@ def ensure_gcp_cloud_creds(config_dir, profile):
     safe_write_to_disk(credentials_file_loc, json.dumps(credentials_json))
 
 
-def ensure_aws_cloud_creds(config_dir, profile):
+def ensure_aws_cloud_creds(config_dir, profile, cspr_override):
     token_info = get_aws_auth_credentials(config_dir, profile)
 
     try:
@@ -523,11 +529,11 @@ def ensure_aws_cloud_creds(config_dir, profile):
         "web_identity_token_file": token_file_loc,
     }
 
-    if token_info.get("cspr_role_arn"):
+    if token_info.get("cspr_role_arn") or cspr_override:
         # If CSPR role is present, then we need to use the task role (in the task profile)
         # to assume the CSPR role.
         aws_config["profile outerbounds"] = {
-            "role_arn": token_info["cspr_role_arn"],
+            "role_arn": cspr_override or token_info["cspr_role_arn"],
             "source_profile": "task",
         }
     else:

{outerbounds-0.9.0.dist-info → outerbounds-0.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: outerbounds
-Version: 0.9.0
+Version: 0.9.1
 Summary: More Data Science, Less Administration
 License: Proprietary
 Keywords: data science,machine learning,MLOps
@@ -30,7 +30,7 @@ Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: metaflow_checkpoint (==0.2.4)
 Requires-Dist: ob-metaflow (==2.18.0.1)
 Requires-Dist: ob-metaflow-extensions (==1.4.9)
-Requires-Dist: ob-metaflow-stubs (==6.0.9.0)
+Requires-Dist: ob-metaflow-stubs (==6.0.9.1)
 Requires-Dist: ob-project-utils (==0.1.35)
 Requires-Dist: opentelemetry-distro (>=0.41b0) ; extra == "otel"
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.20.0) ; extra == "otel"

{outerbounds-0.9.0.dist-info → outerbounds-0.9.1.dist-info}/RECORD

@@ -48,7 +48,7 @@ outerbounds/command_groups/fast_bakery_cli.py,sha256=5kja7v6C651XAY6dsP_IkBPJQgf
 outerbounds/command_groups/flowprojects_cli.py,sha256=gFAA_zUIyhD092Hd7IW5InuIxOqdwRJsHgyWQjy8LZw,3792
 outerbounds/command_groups/kubernetes_cli.py,sha256=2bxPKUp5g_gdwVo4lT-IeWvHxz6Jmj1KxG70nXNgX_M,14758
 outerbounds/command_groups/local_setup_cli.py,sha256=tuuqJRXQ_guEwOuQSIf9wkUU0yg8yAs31myGViAK15s,36364
-outerbounds/command_groups/perimeters_cli.py,sha256=iF_Uw7ROiSctf6FgoJEy30iDBLVE1j9FKuR3shgJRmc,19050
+outerbounds/command_groups/perimeters_cli.py,sha256=z8vwEpNgRXMIMAWltpdaQD72fYGld-mUkcTHrkmkCyk,19299
 outerbounds/command_groups/secrets_cli.py,sha256=Vgn_aiTo76a0s5hCJhNWEOrCVhyYeivD08ooQxz0y7c,2952
 outerbounds/command_groups/tutorials_cli.py,sha256=UInFyiMqtscHFfi8YQwiY_6Sdw9quJOtRu5OukEBccw,3522
 outerbounds/command_groups/workstations_cli.py,sha256=W3OFmInYUfd7Gl3baXIIleZWx6NI3bSeXAqCZ-uvYfA,27901
@@ -60,7 +60,7 @@ outerbounds/utils/schema.py,sha256=lMUr9kNgn9wy-sO_t_Tlxmbt63yLeN4b0xQXbDUDj4A,2
 outerbounds/utils/ssh_utils.py,sha256=MRJ7Oy0M7Lqqs397JDJczTYvxvyHJ_XvBdjt4ZYKaPE,19599
 outerbounds/utils/utils.py,sha256=4Z8cszNob_8kDYCLNTrP-wWads_S_MdL3Uj3ju4mEsk,501
 outerbounds/vendor.py,sha256=gRLRJNXtZBeUpPEog0LOeIsl6GosaFFbCxUvR4bW6IQ,5093
-outerbounds-0.9.0.dist-info/METADATA,sha256=ZXOyjiIoT1aBkxAV4-eHK4IZLWX0nSTxB5cuMS0FPx0,1873
-outerbounds-0.9.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-outerbounds-0.9.0.dist-info/entry_points.txt,sha256=AP6rZg7y5SK9e9a9iVq0Fi9Q2KPjPZSwtZ6R98rLw-8,56
-outerbounds-0.9.0.dist-info/RECORD,,
+outerbounds-0.9.1.dist-info/METADATA,sha256=LtvEYI1WRfyW1cWu_yEuZp0Jo7xneT0FZKnrfzGIgNw,1873
+outerbounds-0.9.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+outerbounds-0.9.1.dist-info/entry_points.txt,sha256=AP6rZg7y5SK9e9a9iVq0Fi9Q2KPjPZSwtZ6R98rLw-8,56
+outerbounds-0.9.1.dist-info/RECORD,,