chalkpy 2.95.0__tar.gz → 2.95.2__tar.gz

{chalkpy-2.95.0 → chalkpy-2.95.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.95.0
+Version: 2.95.2
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

{chalkpy-2.95.0 → chalkpy-2.95.2}/chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py

@@ -18,7 +18,7 @@ from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05\x32\xab\x01\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
+    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data"\xb5\x02\n\x1cTestStreamingResolverRequest\x12&\n\x0cresolver_fqn\x18\x01 \x01(\tH\x00R\x0bresolverFqn\x88\x01\x01\x12@\n\x1astatic_stream_resolver_b64\x18\x02 \x01(\tH\x01R\x17staticStreamResolverB64\x88\x01\x01\x12\x1d\n\ninput_data\x18\x03 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x04 \x01(\tH\x02R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x03R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_resolver_fqnB\x1d\n\x1b_static_stream_resolver_b64B\x0f\n\r_operation_idB\x08\n\x06_debug"\xe1\x01\n\x1dTestStreamingResolverResponse\x12\x44\n\x06status\x18\x01 \x01(\x0e\x32,.chalk.streaming.v1.TestStreamResolverStatusR\x06status\x12\x1e\n\x08\x64\x61ta_uri\x18\x02 \x01(\tH\x00R\x07\x64\x61taUri\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x18\n\x07message\x18\x04 \x01(\tR\x07messageB\x0b\n\t_data_uri*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05*\x99\x01\n\x18TestStreamResolverStatus\x12+\n\'TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED\x10\x00\x12\'\n#TEST_STREAM_RESOLVER_STATUS_SUCCESS\x10\x01\x12\'\n#TEST_STREAM_RESOLVER_STATUS_FAILURE\x10\x02\x32\xaf\x02\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x12\x81\x01\n\x15TestStreamingResolver\x12\x30.chalk.streaming.v1.TestStreamingResolverRequest\x1a\x31.chalk.streaming.v1.TestStreamingResolverResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -31,14 +31,22 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     ]._serialized_options = b"\n\026com.chalk.streaming.v1B\033SimpleStreamingServiceProtoP\001\242\002\003CSX\252\002\022Chalk.Streaming.V1\312\002\022Chalk\\Streaming\\V1\342\002\036Chalk\\Streaming\\V1\\GPBMetadata\352\002\024Chalk::Streaming::V1"
     _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._options = None
     _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._serialized_options = b"\200}\002"
-    _globals["_EXECUTIONPHASE"]._serialized_start = 884
-    _globals["_EXECUTIONPHASE"]._serialized_end = 1081
+    _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["TestStreamingResolver"]._options = None
+    _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["TestStreamingResolver"]._serialized_options = b"\200}\002"
+    _globals["_EXECUTIONPHASE"]._serialized_start = 1424
+    _globals["_EXECUTIONPHASE"]._serialized_end = 1621
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_start = 1624
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_end = 1777
     _globals["_STREAMINGERROR"]._serialized_start = 142
     _globals["_STREAMINGERROR"]._serialized_end = 302
     _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_start = 305
     _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_end = 519
     _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start = 522
     _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end = 881
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1084
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 1255
+    _globals["_TESTSTREAMINGRESOLVERREQUEST"]._serialized_start = 884
+    _globals["_TESTSTREAMINGRESOLVERREQUEST"]._serialized_end = 1193
+    _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_start = 1196
+    _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_end = 1421
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1780
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 2083
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.95.0 → chalkpy-2.95.2}/chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi

@@ -23,12 +23,21 @@ class ExecutionPhase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     EXECUTION_PHASE_AGGREGATION: _ClassVar[ExecutionPhase]
     EXECUTION_PHASE_SINK: _ClassVar[ExecutionPhase]
 
+class TestStreamResolverStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: _ClassVar[TestStreamResolverStatus]
+    TEST_STREAM_RESOLVER_STATUS_SUCCESS: _ClassVar[TestStreamResolverStatus]
+    TEST_STREAM_RESOLVER_STATUS_FAILURE: _ClassVar[TestStreamResolverStatus]
+
 EXECUTION_PHASE_UNSPECIFIED: ExecutionPhase
 EXECUTION_PHASE_PARSE: ExecutionPhase
 EXECUTION_PHASE_MAPPING: ExecutionPhase
 EXECUTION_PHASE_PERSISTENCE: ExecutionPhase
 EXECUTION_PHASE_AGGREGATION: ExecutionPhase
 EXECUTION_PHASE_SINK: ExecutionPhase
+TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: TestStreamResolverStatus
+TEST_STREAM_RESOLVER_STATUS_SUCCESS: TestStreamResolverStatus
+TEST_STREAM_RESOLVER_STATUS_FAILURE: TestStreamResolverStatus
 
 class StreamingError(_message.Message):
     __slots__ = ("error", "phase", "operation_id")
@@ -86,3 +95,42 @@ class SimpleStreamingUnaryInvokeResponse(_message.Message):
         output_data: _Optional[bytes] = ...,
         execution_errors: _Optional[_Iterable[_Union[StreamingError, _Mapping]]] = ...,
     ) -> None: ...
+
+class TestStreamingResolverRequest(_message.Message):
+    __slots__ = ("resolver_fqn", "static_stream_resolver_b64", "input_data", "operation_id", "debug")
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    STATIC_STREAM_RESOLVER_B64_FIELD_NUMBER: _ClassVar[int]
+    INPUT_DATA_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    DEBUG_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    static_stream_resolver_b64: str
+    input_data: bytes
+    operation_id: str
+    debug: bool
+    def __init__(
+        self,
+        resolver_fqn: _Optional[str] = ...,
+        static_stream_resolver_b64: _Optional[str] = ...,
+        input_data: _Optional[bytes] = ...,
+        operation_id: _Optional[str] = ...,
+        debug: bool = ...,
+    ) -> None: ...
+
+class TestStreamingResolverResponse(_message.Message):
+    __slots__ = ("status", "data_uri", "errors", "message")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    DATA_URI_FIELD_NUMBER: _ClassVar[int]
+    ERRORS_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    status: TestStreamResolverStatus
+    data_uri: str
+    errors: _containers.RepeatedCompositeFieldContainer[_chalk_error_pb2.ChalkError]
+    message: str
+    def __init__(
+        self,
+        status: _Optional[_Union[TestStreamResolverStatus, str]] = ...,
+        data_uri: _Optional[str] = ...,
+        errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...,
+        message: _Optional[str] = ...,
+    ) -> None: ...

{chalkpy-2.95.0 → chalkpy-2.95.2}/chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py

@@ -22,6 +22,11 @@ class SimpleStreamingServiceStub(object):
             request_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeRequest.SerializeToString,
             response_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeResponse.FromString,
         )
+        self.TestStreamingResolver = channel.unary_unary(
+            "/chalk.streaming.v1.SimpleStreamingService/TestStreamingResolver",
+            request_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.SerializeToString,
+            response_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.FromString,
+        )
 
 
 class SimpleStreamingServiceServicer(object):
@@ -35,6 +40,15 @@ class SimpleStreamingServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def TestStreamingResolver(self, request, context):
+        """Test a streaming resolver with provided messages.
+        Supports testing both deployed resolvers and static/undeployed resolvers.
+        Returns a signed URL to download the test results.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_SimpleStreamingServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -43,6 +57,11 @@ def add_SimpleStreamingServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeRequest.FromString,
             response_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeResponse.SerializeToString,
         ),
+        "TestStreamingResolver": grpc.unary_unary_rpc_method_handler(
+            servicer.TestStreamingResolver,
+            request_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.FromString,
+            response_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
         "chalk.streaming.v1.SimpleStreamingService", rpc_method_handlers
@@ -82,3 +101,32 @@ class SimpleStreamingService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def TestStreamingResolver(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.streaming.v1.SimpleStreamingService/TestStreamingResolver",
+            chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.SerializeToString,
+            chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

{chalkpy-2.95.0 → chalkpy-2.95.2}/chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi

@@ -10,6 +10,8 @@ from abc import (
 from chalk._gen.chalk.streaming.v1.simple_streaming_service_pb2 import (
     SimpleStreamingUnaryInvokeRequest,
     SimpleStreamingUnaryInvokeResponse,
+    TestStreamingResolverRequest,
+    TestStreamingResolverResponse,
 )
 from grpc import (
     Channel,
@@ -27,6 +29,14 @@ class SimpleStreamingServiceStub:
     """Runs a simple streaming plan with the given request.
     This is a simplified version of the streaming invoker service.
     """
+    TestStreamingResolver: UnaryUnaryMultiCallable[
+        TestStreamingResolverRequest,
+        TestStreamingResolverResponse,
+    ]
+    """Test a streaming resolver with provided messages.
+    Supports testing both deployed resolvers and static/undeployed resolvers.
+    Returns a signed URL to download the test results.
+    """
 
 class SimpleStreamingServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -38,5 +48,15 @@ class SimpleStreamingServiceServicer(metaclass=ABCMeta):
         """Runs a simple streaming plan with the given request.
         This is a simplified version of the streaming invoker service.
         """
+    @abstractmethod
+    def TestStreamingResolver(
+        self,
+        request: TestStreamingResolverRequest,
+        context: ServicerContext,
+    ) -> TestStreamingResolverResponse:
+        """Test a streaming resolver with provided messages.
+        Supports testing both deployed resolvers and static/undeployed resolvers.
+        Returns a signed URL to download the test results.
+        """
 
 def add_SimpleStreamingServiceServicer_to_server(servicer: SimpleStreamingServiceServicer, server: Server) -> None: ...

chalkpy-2.95.2/chalk/_version.py

@@ -0,0 +1 @@
+__version__ = "2.95.2"

{chalkpy-2.95.0 → chalkpy-2.95.2}/chalk/client/client.py

@@ -2251,44 +2251,44 @@ class ChalkClient:
 
         Parameters
         ----------
-        name : str
-           Unique name for the model
-        aliases : list of str, optional
-           List of version aliases (e.g., ["v1.0", "latest"])
-        model : object, optional
-           Python model object (for object-based registration)
-        model_paths : list of str, optional
-           Paths to model files (for file-based registration)
-        additional_files : List[str], optional
-           Additional files needed for inference (tokenizers, configs, etc.)
-        model_type : ModelType, optional
-           Type of model framework
-        model_encoding : ModelEncoding, optional
-           Serialization format
-        input_schema : dict, list, or Any
-           Definition of the input schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        output_schema : dict, list, or Any
-           Definition of the output schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        metadata : dict, optional
-           Additional metadata dictionary containing framework info,
-           training details, performance metrics, etc.
-        input_features : FeatureReference, str, optional
+        name
+            Unique name for the model.
+        aliases
+            List of version aliases (e.g., `["v1.0", "latest"]`).
+        model
+            Python model object (for object-based registration).
+        model_paths
+            Paths to model files (for file-based registration).
+        additional_files
+            Additional files needed for inference (tokenizers, configs, etc.)
+        model_type
+            Type of model framework
+        model_encoding
+            Serialization format
+        input_schema
+            Definition of the input schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        output_schema
+            Definition of the output schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        metadata
+            Additional metadata dictionary containing framework info,
+            training details, performance metrics, etc.
+        input_features
             The features to be used as inputs to the model.
             For example, `[User.message]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.message"]`
-        output_features : FeatureReference, str, optional
+            e.g. `["user.message"]`.
+        output_features
             The features to be used as outputs to the model.
             For example, `[User.is_spam]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.is_spam"]`
-        source_config : LocalSourceConfig, S3SourceConfig, HFSourceConfig, optional
+            e.g. `["user.is_spam"]`.
+        source_config
             Config to pass credentials to access files from a remote source.
-        dependencies : List[str], optional
+        dependencies
             List of package dependencies needed to run this model.
-            e.g. ["torch==2.7.1", "numpy==1.26.4"]
+            e.g. `["torch==2.7.1", "numpy==1.26.4"]`.
 
         Returns
         -------