Make lint

Author: ritaagarwala-sf

src/datacustomcode/cli.py

@@ -208,9 +208,9 @@ def deploy(
     network: str,
     sf_cli_org: Optional[str],
 ):
+    from datacustomcode.constants import USE_IN_FEATURE_MAPPING_FOR_CONNECT_API
     from datacustomcode.deploy import (
         COMPUTE_TYPES,
-        USE_IN_FEATURE_MAPPING_FOR_CONNECT_API,
         CodeExtensionMetadata,
         deploy_full,
         infer_use_in_feature,

src/datacustomcode/function_utils.py

@@ -79,7 +79,7 @@ def get_type_name(type_annotation: Any) -> Optional[str]:
         return None
 
     if hasattr(type_annotation, "__name__"):
-        return type_annotation.__name__
+        return str(type_annotation.__name__)
 
     return str(type_annotation)
 
@@ -230,14 +230,14 @@ def inspect_function_types(
         return None, None
 
 
-def get_request_type(entrypoint_path: str) -> Optional[Any]:
+def get_request_type(entrypoint_path: str) -> Any:
     """Get the request type annotation from a function entrypoint.
 
     Args:
         entrypoint_path: Path to the entrypoint.py file
 
     Returns:
-        The request type (Pydantic model class), or None if not found
+        The request type (Pydantic model class)
 
     Raises:
         ImportError: If the module cannot be loaded

src/datacustomcode/template.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 import os
 import shutil
+from typing import Optional
 
 from loguru import logger
 
@@ -39,7 +40,7 @@ def copy_script_template(target_dir: str) -> None:
             shutil.copy2(source, destination)
 
 
-def copy_function_template(target_dir: str, use_in_feature: str) -> None:
+def copy_function_template(target_dir: str, use_in_feature: Optional[str]) -> None:
     os.makedirs(target_dir, exist_ok=True)
 
     # First, copy common files from base function template