Skip to content

Commit

Permalink
Updated stubs
Browse files Browse the repository at this point in the history
  • Loading branch information
dagardner-nv committed Dec 11, 2024
1 parent ace6106 commit 16de572
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -3,5 +3,5 @@ import datetime
import mrc.core.segment
__all__ = ['RabbitMQSourceStage']
class RabbitMQSourceStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, host: str, exchange: str, exchange_type: str = 'fanout', queue_name: str = '', poll_interval: datetime.timedelta = ...) -> None:
def __init__(self, builder: mrc.core.segment.Builder, name: str, host: str, exchange: str, exchange_type: str = 'fanout', queue_name: str = '', poll_interval: datetime.timedelta = datetime.timedelta(microseconds=100000)) -> None:
...
4 changes: 2 additions & 2 deletions python/morpheus/morpheus/_lib/common/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -328,12 +328,12 @@ def determine_file_type(filename: str) -> FileTypes:
@typing.overload
def determine_file_type(filename: os.PathLike) -> FileTypes:
...
def read_file_to_df(filename: str, file_type: FileTypes = ...) -> typing.Any:
def read_file_to_df(filename: str, file_type: FileTypes = FileTypes.Auto) -> typing.Any:
...
def typeid_is_fully_supported(arg0: TypeId) -> bool:
...
def typeid_to_numpy_str(arg0: TypeId) -> str:
...
def write_df_to_file(df: typing.Any, filename: str, file_type: FileTypes = ..., **kwargs) -> None:
def write_df_to_file(df: typing.Any, filename: str, file_type: FileTypes = FileTypes.Auto, **kwargs) -> None:
...
__version__: str = '25.2.0'
6 changes: 3 additions & 3 deletions python/morpheus/morpheus/_lib/stages/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -53,10 +53,10 @@ class KafkaSourceStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, max_batch_size: int, topics: list[str], batch_timeout_ms: int, config: dict[str, str], disable_commits: bool = False, disable_pre_filtering: bool = False, stop_after: int = 0, async_commits: bool = True, oauth_callback: typing.Callable | None = None) -> None:
...
class MonitorControlMessageStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, description: str, unit: str = 'messages', text_color: morpheus._lib.common.IndicatorsTextColor = ..., font_style: morpheus._lib.common.IndicatorsFontStyle = ..., determine_count_fn: typing.Callable[[morpheus._lib.messages.ControlMessage], int] | None = None) -> None:
def __init__(self, builder: mrc.core.segment.Builder, name: str, description: str, unit: str = 'messages', text_color: morpheus._lib.common.IndicatorsTextColor = IndicatorsTextColor.cyan, font_style: morpheus._lib.common.IndicatorsFontStyle = IndicatorsFontStyle.bold, determine_count_fn: typing.Callable[[morpheus._lib.messages.ControlMessage], int] | None = None) -> None:
...
class MonitorMessageMetaStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, description: str, unit: str = 'messages', text_color: morpheus._lib.common.IndicatorsTextColor = ..., font_style: morpheus._lib.common.IndicatorsFontStyle = ..., determine_count_fn: typing.Callable[[morpheus._lib.messages.MessageMeta], int] | None = None) -> None:
def __init__(self, builder: mrc.core.segment.Builder, name: str, description: str, unit: str = 'messages', text_color: morpheus._lib.common.IndicatorsTextColor = IndicatorsTextColor.cyan, font_style: morpheus._lib.common.IndicatorsFontStyle = IndicatorsFontStyle.bold, determine_count_fn: typing.Callable[[morpheus._lib.messages.MessageMeta], int] | None = None) -> None:
...
class PreallocateControlMessageStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, needed_columns: list[tuple[str, morpheus._lib.common.TypeId]]) -> None:
Expand All @@ -74,6 +74,6 @@ class SerializeStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, include: list[str], exclude: list[str], fixed_columns: bool = True) -> None:
...
class WriteToFileStage(mrc.core.segment.SegmentObject):
def __init__(self, builder: mrc.core.segment.Builder, name: str, filename: str, mode: str = 'w', file_type: morpheus._lib.common.FileTypes = ..., include_index_col: bool = True, flush: bool = False) -> None:
def __init__(self, builder: mrc.core.segment.Builder, name: str, filename: str, mode: str = 'w', file_type: morpheus._lib.common.FileTypes = FileTypes.Auto, include_index_col: bool = True, flush: bool = False) -> None:
...
__version__: str = '25.2.0'

0 comments on commit 16de572

Please sign in to comment.