haystack-ml-stack 0.4.8__tar.gz → 0.4.9__tar.gz

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/PKG-INFO

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

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/pyproject.toml

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

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/src/haystack_ml_stack/_serializers.py

@@ -275,6 +275,8 @@ class UserPlaylistStatsSerializerV1(SimpleSerializer):
             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"])
+            category_msg.capped_total_log_watched = float(stats.get("capped_total_log_watched", 0.0))
+            category_msg.first_active_date = str(stats.get("first_active_date", ""))
         return root_msg
 
 

haystack_ml_stack-0.4.9/src/haystack_ml_stack/_version.py

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

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/src/haystack_ml_stack/generated/v1/features_pb2.py

@@ -24,7 +24,7 @@ _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')
+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\"\xc5\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\x12 \n\x18\x63\x61pped_total_log_watched\x18\x06 \x01(\x02\x12\x19\n\x11\x66irst_active_date\x18\x07 \x01(\t\"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)
@@ -54,37 +54,37 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _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
+  _globals['_PLAYLISTSTATSFORUSER']._serialized_end=1395
+  _globals['_CHANNEL']._serialized_start=1397
+  _globals['_CHANNEL']._serialized_end=1464
+  _globals['_STREAMPSELECT']._serialized_start=1466
+  _globals['_STREAMPSELECT']._serialized_end=1573
+  _globals['_STREAMPWATCHED']._serialized_start=1575
+  _globals['_STREAMPWATCHED']._serialized_end=1672
+  _globals['_USERPWATCHED']._serialized_start=1674
+  _globals['_USERPWATCHED']._serialized_end=1769
+  _globals['_USERPERSONALIZINGPWATCHED']._serialized_start=1772
+  _globals['_USERPERSONALIZINGPWATCHED']._serialized_end=1990
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_start=1897
+  _globals['_USERPERSONALIZINGPWATCHED_DATAENTRY']._serialized_end=1990
+  _globals['_USERPSELECT']._serialized_start=1992
+  _globals['_USERPSELECT']._serialized_end=2097
+  _globals['_USERPERSONALIZINGPSELECT']._serialized_start=2100
+  _globals['_USERPERSONALIZINGPSELECT']._serialized_end=2327
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_start=2223
+  _globals['_USERPERSONALIZINGPSELECT_DATAENTRY']._serialized_end=2327
+  _globals['_STREAMSIMILARITYSCORES']._serialized_start=2330
+  _globals['_STREAMSIMILARITYSCORES']._serialized_end=2492
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_start=2449
+  _globals['_STREAMSIMILARITYSCORES_DATAENTRY']._serialized_end=2492
+  _globals['_GLOBALPLAYLISTSTATS']._serialized_start=2495
+  _globals['_GLOBALPLAYLISTSTATS']._serialized_end=2703
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._serialized_start=2608
+  _globals['_GLOBALPLAYLISTSTATS_DATAENTRY']._serialized_end=2703
+  _globals['_USERPLAYLISTSTATS']._serialized_start=2706
+  _globals['_USERPLAYLISTSTATS']._serialized_end=2908
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._serialized_start=2815
+  _globals['_USERPLAYLISTSTATS_DATAENTRY']._serialized_end=2908
+  _globals['_GLOBALCHANNELS']._serialized_start=2910
+  _globals['_GLOBALCHANNELS']._serialized_end=2994
 # @@protoc_insertion_point(module_scope)

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/src/haystack_ml_stack/generated/v1/features_pb2.pyi

@@ -81,12 +81,16 @@ class PlaylistStatsForUser(_message.Message):
     ACTIVE_DAYS_FIELD_NUMBER: _ClassVar[int]
     TOTAL_WATCHED_FIELD_NUMBER: _ClassVar[int]
     CAPPED_TOTAL_WATCHED_FIELD_NUMBER: _ClassVar[int]
+    CAPPED_TOTAL_LOG_WATCHED_FIELD_NUMBER: _ClassVar[int]
+    FIRST_ACTIVE_DATE_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: ...
+    capped_total_log_watched: float
+    first_active_date: str
+    def __init__(self, total_days: _Optional[int] = ..., start_days: _Optional[int] = ..., active_days: _Optional[int] = ..., total_watched: _Optional[float] = ..., capped_total_watched: _Optional[float] = ..., capped_total_log_watched: _Optional[float] = ..., first_active_date: _Optional[str] = ...) -> None: ...
 
 class Channel(_message.Message):
     __slots__ = ()

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/src/haystack_ml_stack/utils.py

@@ -715,6 +715,8 @@ def _complete_features_for_channels(
         "active_days",
         "total_watched",
         "capped_total_watched",
+        "capped_total_log_watched",
+        "first_active_date",
     ]
 
     for ch in channels:

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/src/haystack_ml_stack.egg-info/PKG-INFO

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

{haystack_ml_stack-0.4.8 → haystack_ml_stack-0.4.9}/tests/test_serializers.py

@@ -150,3 +150,64 @@ def test_user_bias_serializers():
         "User pselect serialization does not match expected value."
     )
     return
+
+
+def test_user_playlist_stats_serializer_backward_compatibility():
+    serializer = SerializerRegistry[("USER", "PLAYLIST_CATEGORY_STATS#3M", "v1")]
+
+    # Case 1: true old binary — built without fields 6 and 7, identical to what
+    # the old serializer would have written before the proto change
+    old_msg = features_pb2_v1.UserPlaylistStats()
+    old_msg.version = 1
+    entry = old_msg.data["sports"]
+    entry.total_days = 30
+    entry.start_days = 5
+    entry.active_days = 10
+    entry.total_watched = 120.5
+    entry.capped_total_watched = 100.0
+    old_binary = old_msg.SerializeToString()
+
+    result = serializer.deserialize(old_binary)
+    assert result.data["sports"].total_days == 30
+    assert result.data["sports"].capped_total_watched == 100.0
+    assert result.data["sports"].capped_total_log_watched == 0.0
+    assert result.data["sports"].first_active_date == ""
+
+    # Case 2: new data with new fields — full round-trip
+    new_data = {
+        "version": 1,
+        "data": {
+            "sports": {
+                "total_days": 30,
+                "start_days": 5,
+                "active_days": 10,
+                "total_watched": 120.5,
+                "capped_total_watched": 100.0,
+                "capped_total_log_watched": 4.8,
+                "first_active_date": "2024-01-15",
+            }
+        },
+    }
+    binary = serializer.serialize(new_data)
+    result = serializer.deserialize(binary)
+    assert abs(result.data["sports"].capped_total_log_watched - 4.8) < 1e-6
+    assert result.data["sports"].first_active_date == "2024-01-15"
+
+    # Case 3: old-format dict missing the new fields — .get() fallback
+    old_format_data = {
+        "version": 1,
+        "data": {
+            "sports": {
+                "total_days": 30,
+                "start_days": 5,
+                "active_days": 10,
+                "total_watched": 120.5,
+                "capped_total_watched": 100.0,
+            }
+        },
+    }
+    binary = serializer.serialize(old_format_data)
+    result = serializer.deserialize(binary)
+    assert result.data["sports"].total_days == 30
+    assert result.data["sports"].capped_total_log_watched == 0.0
+    assert result.data["sports"].first_active_date == ""

haystack_ml_stack-0.4.8/src/haystack_ml_stack/_version.py

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