update

Author: ppandit-sfdc

README.md

@@ -144,7 +144,7 @@ Options:
 - `--name TEXT`: Name of the transformation job [required]
 - `--version TEXT`: Version of the transformation job (default: "0.0.1")
 - `--description TEXT`: Description of the transformation job (default: "")
-- `--compute-type TEXT`: Compute type for the deployment (default: "CPU_M"). Available options: CPU_XS (extra-small), CPU_S (small), CPU_M (medium), CPU_L (large)
+- `--compute-type TEXT`: Compute type for the deployment (default: "CPU_XL"). Available options: CPU_L(Large), CPU_XL (Extra Large), CPU_2XL (2X Large), CPU_4XL (4X Large)
 
 #### `datacustomcode init`
 Initialize a new development environment with a template.

src/datacustomcode/cli.py

@@ -83,7 +83,7 @@ def zip(path: str):
 @click.option("--name", required=True)
 @click.option("--version", default="0.0.1")
 @click.option("--description", default="Custom Data Transform Code")
-@click.option("--compute-type", default="CPU_M")
+@click.option("--compute-type", default="CPU_XL")
 def deploy(path: str, name: str, version: str, description: str, compute_type: str):
     from datacustomcode.credentials import Credentials
     from datacustomcode.deploy import TransformationJobMetadata, deploy_full
@@ -106,7 +106,7 @@ def deploy(path: str, name: str, version: str, description: str, compute_type: s
         name=name,
         version=version,
         description=description,
-        computeType=compute_type,
+        computeType=COMPUTE_TYPES[compute_type],
     )
     try:
         credentials = Credentials.from_available()

src/datacustomcode/deploy.py

@@ -44,20 +44,20 @@
 AUTH_PATH = "services/oauth2/token"
 WAIT_FOR_DEPLOYMENT_TIMEOUT = 3000
 
-# Available compute types for Data Cloud deployments
+# Available compute types for Data Cloud deployments.
 COMPUTE_TYPES = {
-    "CPU_XS": "CPU_XS",      # Extra Small CPU instance
-    "CPU_S": "CPU_S",      # Small CPU instance
-    "CPU_M": "CPU_M",     # Medium CPU instance (default)
-    "CPU_L": "CPU_L"      # Large CPU instance
+    "CPU_L": "CPU_XS",      # Large CPU instance
+    "CPU_XL": "CPU_S",      # X-Large CPU instance
+    "CPU_2XL": "CPU_M",     # 2X-Large CPU instance (default)
+    "CPU_4XL": "CPU_L"      # 4X-Large CPU instance
 }
 
 
 class TransformationJobMetadata(BaseModel):
     name: str
     version: str
     description: str
-    computeType: str = "CPU_M"  # Default to CPU_M for backward compatibility
+    computeType: str
 
     def __init__(self, **data):
         super().__init__(**data)