haystack-ml-stack 0.3.2__tar.gz → 0.3.3__tar.gz

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/PKG-INFO

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

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/pyproject.toml

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

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/src/haystack_ml_stack/__init__.py

@@ -11,4 +11,4 @@ from ._serializers import SerializerRegistry, FeatureRegistryId
 
 __all__ = [*__all__, "SerializerRegistry", "FeatureRegistryId"]
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/src/haystack_ml_stack/app.py

@@ -6,6 +6,7 @@ from http import HTTPStatus
 from typing import Any, Dict, List, Optional
 import time
 from contextlib import asynccontextmanager, AsyncExitStack
+import traceback
 
 import aiobotocore.session
 from aiobotocore.config import AioConfig
@@ -253,7 +254,7 @@ def create_app(
             model_output = model["predict"](model_input, model["params"])
             predict_end = time.perf_counter_ns()
         except Exception as e:
-            logger.error("Model prediction failed: %s", e)
+            logger.error("Model prediction failed: \n%s", traceback.format_exc())
             raise HTTPException(
                 status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
                 detail="Model prediction failed",

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/src/haystack_ml_stack/utils.py

@@ -204,7 +204,7 @@ def device_watched_count_cleanups(
 
     counts_obj: StreamPWatched = stream.get(
         f"PWATCHED#24H#{device_type}", StreamPWatched()
-    )
+    ).data
     if out is None:
         out = {}
     out = _cleanup_entry_context_counts(

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/src/haystack_ml_stack.egg-info/PKG-INFO

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

{haystack_ml_stack-0.3.2 → haystack_ml_stack-0.3.3}/tests/test_utils.py

@@ -372,6 +372,58 @@ def test_stream_pwatched_cleanup():
     assert (actual == expected).all()
 
 
+def test_device_stream_pwatched_cleanup():
+    stream_pwatched_data = {
+        "version": 1,
+        "data": {
+            "autoplay": {"attempts": 1, "watched": 1},
+            "ch_swtch": {"attempts": 2, "watched": 0},
+        },
+    }
+    stream_pwatched = features_pb2_v1.StreamPWatched()
+    ProtoParseDict(js_dict=stream_pwatched_data, message=stream_pwatched)
+    stream = {"PWATCHED#24H#TV": stream_pwatched}
+    out = {}
+    utils.device_watched_count_cleanups(
+        stream=stream,
+        entry_contexts=["autoplay", "ch_swtch", "sel_thumb"],
+        device_type="TV",
+        out=out,
+    )
+    expected = pd.Series(
+        {
+            "STREAM_AUTOPLAY_TV_24H_TOTAL_ATTEMPTS": 1,
+            "STREAM_AUTOPLAY_TV_24H_TOTAL_WATCHED": 1,
+            "STREAM_CH_SWTCH_TV_24H_TOTAL_ATTEMPTS": 2,
+            "STREAM_CH_SWTCH_TV_24H_TOTAL_WATCHED": 0,
+            "STREAM_SEL_THUMB_TV_24H_TOTAL_ATTEMPTS": 0,
+            "STREAM_SEL_THUMB_TV_24H_TOTAL_WATCHED": 0,
+        }
+    )
+    actual = pd.Series(out).loc[expected.index]
+    assert (actual == expected).all()
+    stream = {"PWATCHED#24H#MOBILE": stream_pwatched}
+    out = {}
+    utils.device_watched_count_cleanups(
+        stream=stream,
+        entry_contexts=["autoplay", "ch_swtch", "sel_thumb"],
+        device_type="MOBILE",
+        out=out,
+    )
+    expected = pd.Series(
+        {
+            "STREAM_AUTOPLAY_MOBILE_24H_TOTAL_ATTEMPTS": 1,
+            "STREAM_AUTOPLAY_MOBILE_24H_TOTAL_WATCHED": 1,
+            "STREAM_CH_SWTCH_MOBILE_24H_TOTAL_ATTEMPTS": 2,
+            "STREAM_CH_SWTCH_MOBILE_24H_TOTAL_WATCHED": 0,
+            "STREAM_SEL_THUMB_MOBILE_24H_TOTAL_ATTEMPTS": 0,
+            "STREAM_SEL_THUMB_MOBILE_24H_TOTAL_WATCHED": 0,
+        }
+    )
+    actual = pd.Series(out).loc[expected.index]
+    assert (actual == expected).all()
+
+
 def test_stream_global_pselect_cleanup():
     stream_pselect_data = {
         "version": 1,