fix: ruff

src/codegen/cli/api/client.py

@@ -187,15 +187,19 @@ def identify(self) -> IdentifyResponse | None:
             IdentifyResponse,
         )
 
-    def deploy(self, codemod_name: str, codemod_source: str, lint_mode: bool = False, lint_user_whitelist: list[str] | None = None, message: str | None = None, arguments_schema: dict | None = None) -> DeployResponse:
+    def deploy(self, codemod_name: str, 
+                codemod_source: str, lint_mode: bool = False, 
+                lint_user_whitelist: list[str] | None = None,
+                message: str | None = None, 
+                arguments_schema: dict | None = None) -> DeployResponse:
         """Deploy a codemod to the Modal backend."""
         session = CodegenSession()
         return self._make_request(
             "POST",
             DEPLOY_ENDPOINT,
             DeployInput(
                 input=DeployInput.BaseDeployInput(
-                    codemod_name=codemod_name, codemod_source=codemod_source, repo_full_name=session.repo_name, lint_mode=lint_mode, lint_user_whitelist=lint_user_whitelist or [], message=message
+                    codemod_name=codemod_name, codemod_source=codemod_source, repo_full_name=session.repo_name, lint_mode=lint_mode, lint_user_whitelist=lint_user_whitelist or [], message=message, arguments_schema=arguments_schema
                 )
             ),
             DeployResponse,

src/codegen/cli/utils/json_schema.py

@@ -1,18 +1,15 @@
-# This utility contains functions for utilizing, transforming and validating JSON schemas generated by Pydantic models.
-
 import json
-
 from pydantic import BaseModel
+from pathlib import Path
+from tempfile import TemporaryDirectory
+from datamodel_code_generator import DataModelType, InputFileType, generate
+
+# This utility contains functions for utilizing, transforming and validating JSON schemas generated by Pydantic models.
 
 
 def get_schema(model: BaseModel) -> dict:
     return model.model_json_schema()
 
-from pathlib import Path
-from tempfile import TemporaryDirectory
-
-from datamodel_code_generator import DataModelType, InputFileType, generate
-
 
 def validate_json(schema: dict, json_data: str) -> bool:
     json_schema = json.dumps(schema)