haystack-ml-stack 0.4.2__tar.gz → 0.4.4__tar.gz

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.4.2
+Version: 0.4.4
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "haystack-ml-stack"
-version = "0.4.2"
+version = "0.4.4"
 description = "Functions related to Haystack ML"
 readme = "README.md"
 authors = [{ name = "Oscar Vega", email = "oscar@haystack.tv" }]

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/src/haystack_ml_stack/_kafka.py

@@ -39,6 +39,7 @@ async def send_to_kafka(
         "meta": {
             "monitoring": monitoring_meta,
             "haystack_ml_stack_version": __version__,
+            "playlist_id": str(playlist.get("playlistId", "")),
             "playlist_category": playlist.get("category"),
             "user_features": state.get("user_features", []),
             "stream_features": state.get("stream_features", []),

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/src/haystack_ml_stack/_serializers.py

@@ -231,6 +231,73 @@ class UserPersonalizingPSelectSerializerV1(SimpleSerializer):
         return root_msg.SerializeToString()
 
 
+class GlobalPlaylistStatsSerializerV1(SimpleSerializer):
+    def __init__(self):
+        super().__init__(msg_class=features_pb2_v1.GlobalPlaylistStats)
+
+    def serialize(self, value: dict) -> bytes:
+        root_msg = self.build_msg(value)
+        return root_msg.SerializeToString()
+
+    def build_msg(self, value) -> features_pb2_v1.GlobalPlaylistStats:
+        root_msg = features_pb2_v1.GlobalPlaylistStats()
+        assert value["version"] == 1, "Wrong version given!"
+        root_msg.version = value["version"]
+        data = value["data"]
+        for category, stats in data.items():
+            category_msg = root_msg.data[category]
+            category_msg.watched_count = int(stats["watched_count"])
+            category_msg.not_watched_count = int(stats["not_watched_count"])
+            category_msg.capped_watched_secs = float(stats["capped_watched_secs"])
+            category_msg.capped_not_watched_secs = float(stats["capped_not_watched_secs"])
+            category_msg.watched_secs = float(stats["watched_secs"])
+            category_msg.not_watched_secs = float(stats["not_watched_secs"])
+        return root_msg
+
+
+class UserPlaylistStatsSerializerV1(SimpleSerializer):
+    def __init__(self):
+        super().__init__(msg_class=features_pb2_v1.UserPlaylistStats)
+
+    def serialize(self, value: dict) -> bytes:
+        root_msg = self.build_msg(value)
+        return root_msg.SerializeToString()
+
+    def build_msg(self, value) -> features_pb2_v1.UserPlaylistStats:
+        root_msg = features_pb2_v1.UserPlaylistStats()
+        assert value["version"] == 1, "Wrong version given!"
+        root_msg.version = value["version"]
+        data = value["data"]
+        for category, stats in data.items():
+            category_msg = root_msg.data[category]
+            category_msg.total_days = int(stats["total_days"])
+            category_msg.start_days = int(stats["start_days"])
+            category_msg.active_days = int(stats["active_days"])
+            category_msg.total_watched = float(stats["total_watched"])
+            category_msg.capped_total_watched = float(stats["capped_total_watched"])
+        return root_msg
+
+
+class GlobalChannelsSerializerV1(SimpleSerializer):
+    def __init__(self):
+        super().__init__(msg_class=features_pb2_v1.GlobalChannels)
+
+    def serialize(self, value: dict) -> bytes:
+        root_msg = self.build_msg(value)
+        return root_msg.SerializeToString()
+
+    def build_msg(self, value) -> features_pb2_v1.GlobalChannels:
+        root_msg = features_pb2_v1.GlobalChannels()
+        assert value["version"] == 1, "Wrong version given!"
+        root_msg.version = value["version"]
+        for channel in value["data"]:
+            channel_msg = root_msg.data.add()
+            channel_msg.name = str(channel["name"])
+            channel_msg.category_group = str(channel["category_group"])
+            channel_msg.start_date = int(channel["start_date"])
+        return root_msg
+
+
 class PassThroughSerializer(Serializer):
     def serialize(self, value):
         return value
@@ -249,10 +316,13 @@ stream_pselect_serializer_v0 = StreamPSelectSerializerV0()
 stream_pselect_serializer_v1 = StreamPSelectSerializerV1()
 stream_similarity_scores_serializer_v0 = StreamSimilaritySerializerV0()
 stream_similarity_scores_serializer_v1 = StreamSimilaritySerializerV1()
+global_playlist_stats_serializer_v1 = GlobalPlaylistStatsSerializerV1()
+user_playlist_stats_serializer_v1 = UserPlaylistStatsSerializerV1()
+global_channels_serializer_v1 = GlobalChannelsSerializerV1()
 
 
 class FeatureRegistryId(_t.NamedTuple):
-    entity_type: _t.Literal["STREAM", "USER"]
+    entity_type: _t.Literal["STREAM", "USER", "GLOBAL"]
     feature_id: str
     version: str
 
@@ -341,6 +411,35 @@ user_bias_pselect_v1_features: list[FeatureRegistryId] = [
     FeatureRegistryId(entity_type="USER", feature_id="PSELECT#6M", version="v1")
 ]
 
+global_playlist_stats_v1_features: list[FeatureRegistryId] = [
+    FeatureRegistryId(
+        entity_type="GLOBAL",
+        feature_id="PLAYLIST_CATEGORY_STATS#1D#MOBILE",
+        version="v1",
+    ),
+    FeatureRegistryId(
+        entity_type="GLOBAL",
+        feature_id="PLAYLIST_CATEGORY_STATS#1D#TV",
+        version="v1",
+    ),
+]
+
+user_playlist_stats_v1_features: list[FeatureRegistryId] = [
+    FeatureRegistryId(
+        entity_type="USER",
+        feature_id="PLAYLIST_CATEGORY_STATS#3M",
+        version="v1",
+    ),
+]
+
+global_channels_v1_features: list[FeatureRegistryId] = [
+    FeatureRegistryId(
+        entity_type="GLOBAL",
+        feature_id="CHANNEL_CANDIDATES",
+        version="v1",
+    ),
+]
+
 features_serializer_tuples: list[tuple[list[FeatureRegistryId], Serializer]] = [
     (stream_pwatched_v0_features, stream_pwatched_serializer_v0),
     (stream_pwatched_v1_features, stream_pwatched_serializer_v1),
@@ -355,6 +454,18 @@ features_serializer_tuples: list[tuple[list[FeatureRegistryId], Serializer]] = [
     (user_bias_pwatched_v1_features, user_pwatched_serializer_v1),
     (user_personalizing_pselect_v1_features, user_personalizing_pselect_serializer_v1),
     (user_bias_pselect_v1_features, user_pselect_serializer_v1),
+    (
+        global_playlist_stats_v1_features,
+        global_playlist_stats_serializer_v1,
+    ),
+    (
+        user_playlist_stats_v1_features,
+        user_playlist_stats_serializer_v1,
+    ),
+    (
+        global_channels_v1_features,
+        global_channels_serializer_v1,
+    ),
 ]
 
 SerializerRegistry: dict[FeatureRegistryId, Serializer] = {

haystack_ml_stack-0.4.4/src/haystack_ml_stack/_version.py

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

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/src/haystack_ml_stack/dynamo.py

@@ -150,7 +150,6 @@ async def set_all_features(
     user_cache_misses = len(cache_miss) - stream_cache_misses
     valid_cache_delays = list(v for v in cache_delay_obj.values() if v > 0)
     cache_delay = min(valid_cache_delays) if valid_cache_delays else 0
-
     if not cache_miss:
         return FeatureRetrievalMeta(
             user_cache_misses=0,

haystack_ml_stack-0.4.4/src/haystack_ml_stack/generated/v1/features_pb2.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: features.proto
+# Protobuf Python Version: 6.33.2
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    33,
+    2,
+    '',
+    'features.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x66\x65\x61tures.proto\x12\x1ahaystack_ml_stack.features\"7\n\x12\x45ntryContextCounts\x12\x10\n\x08\x61ttempts\x18\x01 \x01(\x05\x12\x0f\n\x07watched\x18\x02 \x01(\x05\"_\n\x0cSelectCounts\x12\x15\n\rtotal_selects\x18\x01 \x01(\x05\x12!\n\x19total_selects_and_watched\x18\x02 \x01(\x05\x12\x15\n\rtotal_browsed\x18\x03 \x01(\x05\"\xf3\x02\n\x14\x45ntryContextPWatched\x12@\n\x08\x61utoplay\x18\x01 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12\x41\n\tsel_thumb\x18\x02 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12\x43\n\x0b\x63hoose_next\x18\x03 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12@\n\x08\x63h_swtch\x18\x04 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12O\n\x17launch_first_in_session\x18\x05 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\"\x85\x02\n\x0fPositionPSelect\x12;\n\tfirst_pos\x18\x01 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12<\n\nsecond_pos\x18\x02 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12;\n\tthird_pos\x18\x03 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12:\n\x08rest_pos\x18\x04 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\"\xa9\x01\n\x1f\x42rowsedDebiasedPositionPSelects\x12\x44\n\x0fup_to_4_browsed\x18\x01 \x01(\x0b\x32+.haystack_ml_stack.features.PositionPSelect\x12@\n\x0b\x61ll_browsed\x18\x02 \x01(\x0b\x32+.haystack_ml_stack.features.PositionPSelect\"\xb8\x01\n\x16PlaylistStatsForGlobal\x12\x15\n\rwatched_count\x18\x01 \x01(\x05\x12\x19\n\x11not_watched_count\x18\x02 \x01(\x05\x12\x1b\n\x13\x63\x61pped_watched_secs\x18\x03 \x01(\x02\x12\x1f\n\x17\x63\x61pped_not_watched_secs\x18\x04 \x01(\x02\x12\x14\n\x0cwatched_secs\x18\x05 \x01(\x02\x12\x18\n\x10not_watched_secs\x18\x06 \x01(\x02\"\x88\x01\n\x14PlaylistStatsForUser\x12\x12\n\ntotal_days\x18\x01 \x01(\x05\x12\x12\n\nstart_days\x18\x02 \x01(\x05\x12\x13\n\x0b\x61\x63tive_days\x18\x03 \x01(\x05\x12\x15\n\rtotal_watched\x18\x04 \x01(\x02\x12\x1c\n\x14\x63\x61pped_total_watched\x18\x05 \x01(\x02\"C\n\x07\x43hannel\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x16\n\x0e\x63\x61tegory_group\x18\x02 \x01(\t\x12\x12\n\nstart_date\x18\x03 \x01(\x05\"k\n\rStreamPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12I\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects\"a\n\x0eStreamPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12>\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched\"_\n\x0cUserPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12>\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched\"\xda\x01\n\x19UserPersonalizingPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12M\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32?.haystack_ml_stack.features.UserPersonalizingPWatched.DataEntry\x1a]\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12?\n\x05value\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched:\x02\x38\x01\"i\n\x0bUserPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12I\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects\"\xe3\x01\n\x18UserPersonalizingPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12L\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32>.haystack_ml_stack.features.UserPersonalizingPSelect.DataEntry\x1ah\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects:\x02\x38\x01\"\xa2\x01\n\x16StreamSimilarityScores\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12J\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32<.haystack_ml_stack.features.StreamSimilarityScores.DataEntry\x1a+\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\xd0\x01\n\x13GlobalPlaylistStats\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12G\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32\x39.haystack_ml_stack.features.GlobalPlaylistStats.DataEntry\x1a_\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x41\n\x05value\x18\x02 \x01(\x0b\x32\x32.haystack_ml_stack.features.PlaylistStatsForGlobal:\x02\x38\x01\"\xca\x01\n\x11UserPlaylistStats\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12\x45\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32\x37.haystack_ml_stack.features.UserPlaylistStats.DataEntry\x1a]\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12?\n\x05value\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.PlaylistStatsForUser:\x02\x38\x01\"T\n\x0eGlobalChannels\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12\x31\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32#.haystack_ml_stack.features.Channelb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'features_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._loaded_options = None
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_options = b'8\001'
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._loaded_options = None
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_options = b'8\001'
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._loaded_options = None
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_options = b'8\001'
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._loaded_options = None
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._serialized_options = b'8\001'
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._loaded_options = None
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._serialized_options = b'8\001'
+  _globals['_ENTRYCONTEXTCOUNTS']._serialized_start=46
+  _globals['_ENTRYCONTEXTCOUNTS']._serialized_end=101
+  _globals['_SELECTCOUNTS']._serialized_start=103
+  _globals['_SELECTCOUNTS']._serialized_end=198
+  _globals['_ENTRYCONTEXTPWATCHED']._serialized_start=201
+  _globals['_ENTRYCONTEXTPWATCHED']._serialized_end=572
+  _globals['_POSITIONPSELECT']._serialized_start=575
+  _globals['_POSITIONPSELECT']._serialized_end=836
+  _globals['_BROWSEDDEBIASEDPOSITIONPSELECTS']._serialized_start=839
+  _globals['_BROWSEDDEBIASEDPOSITIONPSELECTS']._serialized_end=1008
+  _globals['_PLAYLISTSTATSFORGLOBAL']._serialized_start=1011
+  _globals['_PLAYLISTSTATSFORGLOBAL']._serialized_end=1195
+  _globals['_PLAYLISTSTATSFORUSER']._serialized_start=1198
+  _globals['_PLAYLISTSTATSFORUSER']._serialized_end=1334
+  _globals['_CHANNEL']._serialized_start=1336
+  _globals['_CHANNEL']._serialized_end=1403
+  _globals['_STREAMPSELECT']._serialized_start=1405
+  _globals['_STREAMPSELECT']._serialized_end=1512
+  _globals['_STREAMPWATCHED']._serialized_start=1514
+  _globals['_STREAMPWATCHED']._serialized_end=1611
+  _globals['_USERPWATCHED']._serialized_start=1613
+  _globals['_USERPWATCHED']._serialized_end=1708
+  _globals['_USERPERSONALIZINGPWATCHED']._serialized_start=1711
+  _globals['_USERPERSONALIZINGPWATCHED']._serialized_end=1929
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_start=1836
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_end=1929
+  _globals['_USERPSELECT']._serialized_start=1931
+  _globals['_USERPSELECT']._serialized_end=2036
+  _globals['_USERPERSONALIZINGPSELECT']._serialized_start=2039
+  _globals['_USERPERSONALIZINGPSELECT']._serialized_end=2266
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_start=2162
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_end=2266
+  _globals['_STREAMSIMILARITYSCORES']._serialized_start=2269
+  _globals['_STREAMSIMILARITYSCORES']._serialized_end=2431
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_start=2388
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_end=2431
+  _globals['_GLOBALPLAYLISTSTATS']._serialized_start=2434
+  _globals['_GLOBALPLAYLISTSTATS']._serialized_end=2642
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._serialized_start=2547
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._serialized_end=2642
+  _globals['_USERPLAYLISTSTATS']._serialized_start=2645
+  _globals['_USERPLAYLISTSTATS']._serialized_end=2847
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._serialized_start=2754
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._serialized_end=2847
+  _globals['_GLOBALCHANNELS']._serialized_start=2849
+  _globals['_GLOBALCHANNELS']._serialized_end=2933
+# @@protoc_insertion_point(module_scope)

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/src/haystack_ml_stack/generated/v1/features_pb2.pyi

@@ -1,7 +1,7 @@
 from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from collections.abc import Mapping as _Mapping
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
 from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
@@ -58,6 +58,46 @@ class BrowsedDebiasedPositionPSelects(_message.Message):
     all_browsed: PositionPSelect
     def __init__(self, up_to_4_browsed: _Optional[_Union[PositionPSelect, _Mapping]] = ..., all_browsed: _Optional[_Union[PositionPSelect, _Mapping]] = ...) -> None: ...
 
+class PlaylistStatsForGlobal(_message.Message):
+    __slots__ = ()
+    WATCHED_COUNT_FIELD_NUMBER: _ClassVar[int]
+    NOT_WATCHED_COUNT_FIELD_NUMBER: _ClassVar[int]
+    CAPPED_WATCHED_SECS_FIELD_NUMBER: _ClassVar[int]
+    CAPPED_NOT_WATCHED_SECS_FIELD_NUMBER: _ClassVar[int]
+    WATCHED_SECS_FIELD_NUMBER: _ClassVar[int]
+    NOT_WATCHED_SECS_FIELD_NUMBER: _ClassVar[int]
+    watched_count: int
+    not_watched_count: int
+    capped_watched_secs: float
+    capped_not_watched_secs: float
+    watched_secs: float
+    not_watched_secs: float
+    def __init__(self, watched_count: _Optional[int] = ..., not_watched_count: _Optional[int] = ..., capped_watched_secs: _Optional[float] = ..., capped_not_watched_secs: _Optional[float] = ..., watched_secs: _Optional[float] = ..., not_watched_secs: _Optional[float] = ...) -> None: ...
+
+class PlaylistStatsForUser(_message.Message):
+    __slots__ = ()
+    TOTAL_DAYS_FIELD_NUMBER: _ClassVar[int]
+    START_DAYS_FIELD_NUMBER: _ClassVar[int]
+    ACTIVE_DAYS_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_WATCHED_FIELD_NUMBER: _ClassVar[int]
+    CAPPED_TOTAL_WATCHED_FIELD_NUMBER: _ClassVar[int]
+    total_days: int
+    start_days: int
+    active_days: int
+    total_watched: float
+    capped_total_watched: float
+    def __init__(self, total_days: _Optional[int] = ..., start_days: _Optional[int] = ..., active_days: _Optional[int] = ..., total_watched: _Optional[float] = ..., capped_total_watched: _Optional[float] = ...) -> None: ...
+
+class Channel(_message.Message):
+    __slots__ = ()
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    CATEGORY_GROUP_FIELD_NUMBER: _ClassVar[int]
+    START_DATE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    category_group: str
+    start_date: int
+    def __init__(self, name: _Optional[str] = ..., category_group: _Optional[str] = ..., start_date: _Optional[int] = ...) -> None: ...
+
 class StreamPSelect(_message.Message):
     __slots__ = ()
     VERSION_FIELD_NUMBER: _ClassVar[int]
@@ -134,3 +174,41 @@ class StreamSimilarityScores(_message.Message):
     version: int
     data: _containers.ScalarMap[str, float]
     def __init__(self, version: _Optional[int] = ..., data: _Optional[_Mapping[str, float]] = ...) -> None: ...
+
+class GlobalPlaylistStats(_message.Message):
+    __slots__ = ()
+    class DataEntry(_message.Message):
+        __slots__ = ()
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: PlaylistStatsForGlobal
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[PlaylistStatsForGlobal, _Mapping]] = ...) -> None: ...
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    version: int
+    data: _containers.MessageMap[str, PlaylistStatsForGlobal]
+    def __init__(self, version: _Optional[int] = ..., data: _Optional[_Mapping[str, PlaylistStatsForGlobal]] = ...) -> None: ...
+
+class UserPlaylistStats(_message.Message):
+    __slots__ = ()
+    class DataEntry(_message.Message):
+        __slots__ = ()
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: PlaylistStatsForUser
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[PlaylistStatsForUser, _Mapping]] = ...) -> None: ...
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    version: int
+    data: _containers.MessageMap[str, PlaylistStatsForUser]
+    def __init__(self, version: _Optional[int] = ..., data: _Optional[_Mapping[str, PlaylistStatsForUser]] = ...) -> None: ...
+
+class GlobalChannels(_message.Message):
+    __slots__ = ()
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    version: int
+    data: _containers.RepeatedCompositeFieldContainer[Channel]
+    def __init__(self, version: _Optional[int] = ..., data: _Optional[_Iterable[_Union[Channel, _Mapping]]] = ...) -> None: ...

{haystack_ml_stack-0.4.2 → haystack_ml_stack-0.4.4}/src/haystack_ml_stack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.4.2
+Version: 0.4.4
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT

haystack_ml_stack-0.4.2/src/haystack_ml_stack/_version.py

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

haystack_ml_stack-0.4.2/src/haystack_ml_stack/generated/v1/features_pb2.py

@@ -1,70 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: features.proto
-# Protobuf Python Version: 6.33.2
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    33,
-    2,
-    '',
-    'features.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x66\x65\x61tures.proto\x12\x1ahaystack_ml_stack.features\"7\n\x12\x45ntryContextCounts\x12\x10\n\x08\x61ttempts\x18\x01 \x01(\x05\x12\x0f\n\x07watched\x18\x02 \x01(\x05\"_\n\x0cSelectCounts\x12\x15\n\rtotal_selects\x18\x01 \x01(\x05\x12!\n\x19total_selects_and_watched\x18\x02 \x01(\x05\x12\x15\n\rtotal_browsed\x18\x03 \x01(\x05\"\xf3\x02\n\x14\x45ntryContextPWatched\x12@\n\x08\x61utoplay\x18\x01 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12\x41\n\tsel_thumb\x18\x02 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12\x43\n\x0b\x63hoose_next\x18\x03 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12@\n\x08\x63h_swtch\x18\x04 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\x12O\n\x17launch_first_in_session\x18\x05 \x01(\x0b\x32..haystack_ml_stack.features.EntryContextCounts\"\x85\x02\n\x0fPositionPSelect\x12;\n\tfirst_pos\x18\x01 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12<\n\nsecond_pos\x18\x02 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12;\n\tthird_pos\x18\x03 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\x12:\n\x08rest_pos\x18\x04 \x01(\x0b\x32(.haystack_ml_stack.features.SelectCounts\"\xa9\x01\n\x1f\x42rowsedDebiasedPositionPSelects\x12\x44\n\x0fup_to_4_browsed\x18\x01 \x01(\x0b\x32+.haystack_ml_stack.features.PositionPSelect\x12@\n\x0b\x61ll_browsed\x18\x02 \x01(\x0b\x32+.haystack_ml_stack.features.PositionPSelect\"k\n\rStreamPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12I\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects\"a\n\x0eStreamPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12>\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched\"_\n\x0cUserPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12>\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched\"\xda\x01\n\x19UserPersonalizingPWatched\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12M\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32?.haystack_ml_stack.features.UserPersonalizingPWatched.DataEntry\x1a]\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12?\n\x05value\x18\x02 \x01(\x0b\x32\x30.haystack_ml_stack.features.EntryContextPWatched:\x02\x38\x01\"i\n\x0bUserPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12I\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects\"\xe3\x01\n\x18UserPersonalizingPSelect\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12L\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32>.haystack_ml_stack.features.UserPersonalizingPSelect.DataEntry\x1ah\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.haystack_ml_stack.features.BrowsedDebiasedPositionPSelects:\x02\x38\x01\"\xa2\x01\n\x16StreamSimilarityScores\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12J\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32<.haystack_ml_stack.features.StreamSimilarityScores.DataEntry\x1a+\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'features_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  DESCRIPTOR._loaded_options = None
-  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._loaded_options = None
-  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_options = b'8\001'
-  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._loaded_options = None
-  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_options = b'8\001'
-  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._loaded_options = None
-  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_options = b'8\001'
-  _globals['_ENTRYCONTEXTCOUNTS']._serialized_start=46
-  _globals['_ENTRYCONTEXTCOUNTS']._serialized_end=101
-  _globals['_SELECTCOUNTS']._serialized_start=103
-  _globals['_SELECTCOUNTS']._serialized_end=198
-  _globals['_ENTRYCONTEXTPWATCHED']._serialized_start=201
-  _globals['_ENTRYCONTEXTPWATCHED']._serialized_end=572
-  _globals['_POSITIONPSELECT']._serialized_start=575
-  _globals['_POSITIONPSELECT']._serialized_end=836
-  _globals['_BROWSEDDEBIASEDPOSITIONPSELECTS']._serialized_start=839
-  _globals['_BROWSEDDEBIASEDPOSITIONPSELECTS']._serialized_end=1008
-  _globals['_STREAMPSELECT']._serialized_start=1010
-  _globals['_STREAMPSELECT']._serialized_end=1117
-  _globals['_STREAMPWATCHED']._serialized_start=1119
-  _globals['_STREAMPWATCHED']._serialized_end=1216
-  _globals['_USERPWATCHED']._serialized_start=1218
-  _globals['_USERPWATCHED']._serialized_end=1313
-  _globals['_USERPERSONALIZINGPWATCHED']._serialized_start=1316
-  _globals['_USERPERSONALIZINGPWATCHED']._serialized_end=1534
-  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_start=1441
-  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_end=1534
-  _globals['_USERPSELECT']._serialized_start=1536
-  _globals['_USERPSELECT']._serialized_end=1641
-  _globals['_USERPERSONALIZINGPSELECT']._serialized_start=1644
-  _globals['_USERPERSONALIZINGPSELECT']._serialized_end=1871
-  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_start=1767
-  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_end=1871
-  _globals['_STREAMSIMILARITYSCORES']._serialized_start=1874
-  _globals['_STREAMSIMILARITYSCORES']._serialized_end=2036
-  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_start=1993
-  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_end=2036
-# @@protoc_insertion_point(module_scope)