add flag for profile

Author: joroscoSF

src/datacustomcode/cli.py

@@ -97,7 +97,9 @@ def zip(path: str):
 
     Choose based on your workload requirements.""",
 )
-def deploy(path: str, name: str, version: str, description: str, cpu_size: str, profile: str):
+def deploy(
+    path: str, name: str, version: str, description: str, cpu_size: str, profile: str
+):
     from datacustomcode.credentials import Credentials
     from datacustomcode.deploy import TransformationJobMetadata, deploy_full
 
@@ -194,7 +196,12 @@ def scan(filename: str, config: str, dry_run: bool, no_requirements: bool):
 @click.option("--config-file", default=None)
 @click.option("--dependencies", default=[], multiple=True)
 @click.option("--profile", default="default")
-def run(entrypoint: str, config_file: Union[str, None], dependencies: List[str], profile: str):
+def run(
+    entrypoint: str,
+    config_file: Union[str, None],
+    dependencies: List[str],
+    profile: str,
+):
     from datacustomcode.run import run_entrypoint
 
     run_entrypoint(entrypoint, config_file, dependencies, profile)

src/datacustomcode/run.py

@@ -20,7 +20,10 @@
 
 
 def run_entrypoint(
-    entrypoint: str, config_file: Union[str, None], dependencies: List[str], profile: str
+    entrypoint: str,
+    config_file: Union[str, None],
+    dependencies: List[str],
+    profile: str,
 ) -> None:
     """Run the entrypoint script with the given config and dependencies.
 
@@ -31,11 +34,11 @@ def run_entrypoint(
         profile: The profile to use.
     """
     if profile != "default":
-        if config.reader_config and hasattr(config.reader_config, 'options'):
+        if config.reader_config and hasattr(config.reader_config, "options"):
             config.reader_config.options["credentials_profile"] = profile
-        if config.writer_config and hasattr(config.writer_config, 'options'):
+        if config.writer_config and hasattr(config.writer_config, "options"):
             config.writer_config.options["credentials_profile"] = profile
-    
+
     if config_file:
         config.load(config_file)
     for dependency in dependencies:

tests/test_run.py

@@ -96,6 +96,7 @@ def test_run_entrypoint_preserves_config(test_config_file, test_entrypoint_file)
             entrypoint=test_entrypoint_file,
             config_file=test_config_file,
             dependencies=[],
+            profile="default",
         )
 
         # Check that config was maintained
@@ -180,6 +181,7 @@ def test_run_entrypoint_with_dependencies():
             entrypoint=entrypoint_file,
             config_file=config_file,
             dependencies=[module_name],
+            profile="default",
         )
 
         # Verify dependency was imported and used